Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dpy_checkpoint_reader.py63 def get_tensor(self, tensor_str): function
/aosp_15_r20/external/ComputeLibrary/src/common/
H A DTensorPack.cpp65 arm_compute::ITensor *TensorPack::get_tensor(int32_t slot_id) in get_tensor() function in arm_compute::TensorPack
/aosp_15_r20/external/ComputeLibrary/src/core/
H A DITensorPack.cpp64 ITensor *ITensorPack::get_tensor(int id) in get_tensor() function in arm_compute::ITensorPack
/aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/utils/
H A Dexport.py112 def get_tensor(io_info, tensors, logger, checking_output=False): function
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/
H A DArgumentPack.h123 T *get_tensor(Id id) in get_tensor() function
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dmaia_extension.cpp8 Tensor get_tensor(caffe2::TypeMeta dtype, IntArrayRef size) { in get_tensor() function
/aosp_15_r20/external/pytorch/torch/nn/utils/
H A D_named_member_accessor.py164 def get_tensor(self, name: str) -> torch.Tensor: member in NamedMemberAccessor
/aosp_15_r20/external/executorch/backends/qualcomm/builders/
H A Dnode_visitor.py92 def get_tensor(self, input_node, op_node, idx=None): member in NodeVisitor
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dinterpreter.py837 def get_tensor(self, tensor_index, subgraph_index=0): member in Interpreter
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.h286 virtual TFE_TensorHandle* get_tensor(size_t index) const { in get_tensor() function
H A Ddtensor_device_util.cc496 TFE_TensorHandle* SparseTensorWithLayout::get_tensor(size_t index) const { in get_tensor() function in tensorflow::dtensor::SparseTensorWithLayout
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddebug_data.py338 def get_tensor(self): member in DebugTensorDatum
/aosp_15_r20/external/executorch/exir/emit/test/
H A Dtest_emit.py1178 def get_tensor(self) -> torch.Tensor: member in TestEmit.test_emit_prims.Simple
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_export.py825 def get_tensor(self, i): member in TestExport.test_state_tensors.M
854 def get_tensor(self, i): member in TestExport.test_state_tensors.M