Home
last modified time | relevance | path

Searched defs:tensor_meta (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/executorch/examples/qualcomm/executor_runner/
H A Dqnn_executor_runner.cpp262 Result<TensorInfo> tensor_meta = method_meta.input_tensor_meta(input_index); in main() local
342 Result<TensorInfo> tensor_meta = in main() local
/aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama2/runner/
H A Drunner.cpp292 Result<TensorInfo> tensor_meta = in generate() local
322 Result<TensorInfo> tensor_meta = in generate() local
H A Drunner.h117 std::unique_ptr<executorch::runtime::TensorInfo> tensor_meta; member
/aosp_15_r20/external/executorch/extension/runner_util/
H A Dinputs_aten.cpp27 TensorInfo& tensor_meta, in fill_and_set_input()
H A Dinputs.cpp41 Result<TensorInfo> tensor_meta = method_meta.input_tensor_meta(i); in prepare_input_tensors() local
H A Dinputs_portable.cpp57 TensorInfo& tensor_meta, in fill_and_set_input()
/aosp_15_r20/external/executorch/runtime/executor/
H A Dmethod.cpp837 auto tensor_meta = this->method_meta().input_tensor_meta(input_idx); in set_input() local
962 auto tensor_meta = this->method_meta().output_tensor_meta(output_idx); in set_output_data_ptr() local
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/
H A D_meta_parser.py29 def from_meta(allocator, tensor_meta): argument
/aosp_15_r20/external/pytorch/torch/distributed/tensor/
H A D_dtensor_spec.py30 tensor_meta: Optional[TensorMeta] = None variable in DTensorSpec
/aosp_15_r20/external/executorch/examples/mediatek/executor_runner/
H A Dmtk_oss_executor_runner.cpp233 Result<TensorInfo> tensor_meta = method_meta.input_tensor_meta(i); in main() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dtensor_coding.cc147 protobuf::io::CodedInputStream* input, TensorProto* tensor_meta) { in ParseTensorSubmessage()
/aosp_15_r20/external/executorch/examples/arm/executor_runner/
H A Darm_executor_runner.cpp207 Result<TensorInfo> tensor_meta = method_meta.input_tensor_meta(i); in prepare_input_tensors() local
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dmeta_tracer.py70 def install_tensor_meta(self, tensor_meta): argument
/aosp_15_r20/external/executorch/examples/mediatek/executor_runner/llama_runner/
H A DModelChunk.cpp403 const auto tensor_meta = method_meta.input_tensor_meta(i); in SetBackendInputs() local