Home
last modified time | relevance | path

Searched defs:tensors_ (Results 1 – 25 of 41) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/rpc_messages/
H A Drpc_with_profiling_resp.h53 std::vector<torch::Tensor> tensors_; variable
H A Drpc_with_profiling_req.h56 std::vector<torch::Tensor> tensors_; variable
H A Drpc_with_autograd.h90 std::vector<torch::Tensor> tensors_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dsession_state.h53 std::unordered_map<string, Tensor> tensors_; variable
H A Dvariant_tensor_data.h96 std::vector<Tensor> tensors_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtensor_list.h149 Tensors* tensors_; variable
H A Dtensor_map.h170 Tensors* tensors_; variable
H A Dops_testutil.h178 std::vector<Tensor*> tensors_; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
H A Dop_builder.h91 std::vector<TensorID> tensors_; variable
H A Dutil_test.cc67 std::vector<TfLiteTensor> tensors_; member in __anoncc2fe2500111::IsBinaryOpSupportedTest
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dtypes.h59 std::vector<at::Tensor> tensors_; member
H A Dmessage.h160 std::vector<torch::Tensor> tensors_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dtensor_slice_reader.h135 mutable std::unordered_map<string, TensorSliceSet*> tensors_; variable
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device_lib.h222 const std::vector<TensorHandlePtr> tensors_; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/
H A Dop_signature_test.cc134 std::vector<TfLiteTensor> tensors_; member in tflite::StubTfLiteContext
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dverifier_internal_test.cc134 std::vector<flatbuffers::Offset<Tensor>> tensors_; member in tflite::TfLiteFlatbufferModelBuilder
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dlookup_ops.cc108 std::vector<Tensor> tensors_; member in tensorflow::data::experimental::__anond85f2e6b0111::DatasetIterator
H A Dlist_dataset_op.cc184 const std::vector<Tensor> tensors_; member in tensorflow::data::ListDatasetOp::Dataset
H A Dlist_dataset_op_test.cc110 std::vector<Tensor> tensors_; member in tensorflow::data::__anon29aeafde0111::ListDatasetParams
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dtensor_dataset_op.cc182 const std::vector<Tensor> tensors_; member in tensorflow::data::TensorDatasetOp::Dataset
H A Dtensor_slice_dataset_op.cc205 const std::vector<Tensor> tensors_; member in tensorflow::data::TensorSliceDatasetOp::Dataset
/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Ddata_flow.cpp55 std::vector<RawTensorInfo> tensors_; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dgpu_api_delegate.cc300 std::vector<std::pair<TensorObject, TensorObjectDef>> tensors_; member in tflite::gpu::cl::__anon38a7c2710111::Delegate
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Djit_metaprogramming_utils.py448 def new_fn(*tensors_):
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dop_builder.h394 std::vector<OpBuilder::TensorID> tensors_; variable

12