Tensor Struct Reference

#include <ios_caffe_predictor.h>

Collaboration diagram for Tensor:

Public Attributes
std::vector< int64_t >	dims
uint8_t *	data

Detailed Description

Definition at line 9 of file ios_caffe_predictor.h.

Member Data Documentation

data

uint8_t* Tensor::data

Definition at line 11 of file ios_caffe_predictor.h.

Referenced by torch.nn.parameter.Parameter::__deepcopy__(), torch.nn.parameter.Parameter::__reduce_ex__(), torch.utils.hipify.hipify_python.Trie::add(), caffe2.python.layers.constant_weight.ConstantWeight::add_ops(), torch::jit::tensorexpr::SimpleIREvaluator::bindArg(), caffe2.python.layers.adaptive_weight.AdaptiveWeight::concat_data(), torch::nn::utils::rnn::PackedSequence::cpu(), torch::nn::utils::rnn::PackedSequence::cuda(), torch.utils.hipify.hipify_python.Trie::dump(), operator_benchmark.c2.batch_box_cox_test.BatchBoxCoxBenchmark::forward(), operator_benchmark.c2.quantile_op_test.QuantileOpBenchmark::forward(), conv.ConvImplBench::forward(), normalization.BatchNormBench::forward(), normalization.InstanceNormBench::forward(), normalization.LayerNormBench::forward(), pooling.PoolingBench::forward(), softmax.SoftmaxBench::forward(), common.CommitDataCache::get(), torch::nn::utils::rnn::PackedSequence::is_cuda(), torch::nn::utils::rnn::PackedSequence::is_pinned(), torch.nn.parameter.UninitializedParameter::materialize(), torch::nn::utils::rnn::PackedSequence::pin_memory(), softmax.SoftmaxBench::reference(), swish.SwishBench::reference(), Caffe2IOSPredictor::run(), at::strided_tensor_iter_fixed< T, N >::strided_tensor_iter_fixed(), caffe2.python.db_test.TestDB::testSimple(), THCStorage_resizeBytes(), THTensor_getSizePtr(), THTensor_getStridePtr(), torch::nn::utils::rnn::PackedSequence::to(), common.CommitDataCache::write_to_disk(), and torch::jit::ScriptModuleSerializer::writeArchive().

dims

std::vector<int64_t> Tensor::dims

Definition at line 10 of file ios_caffe_predictor.h.

Referenced by reduction.ReduceBench::forward(), and Caffe2IOSPredictor::run().

---

The documentation for this struct was generated from the following file:

• caffe2/mobile/contrib/ios/ios_caffe_predictor.h