Searched defs:tensor_wrapper (Results 1 – 9 of 9) sorted by relevance
154 const std::shared_ptr<TensorWrapper>& tensor_wrapper) { in RegisterMem()177 const std::shared_ptr<TensorWrapper>& tensor_wrapper) { in RegisterIonMem()213 const std::shared_ptr<TensorWrapper>& tensor_wrapper) { in RegisterCustomMem()335 std::shared_ptr<TensorWrapper> tensor_wrapper = CreateTensorWrapper(tensor); in AllocateTensor() local346 std::shared_ptr<TensorWrapper> tensor_wrapper = in AllocateTensor() local632 for (const auto& tensor_wrapper : op_wrapper->GetInputTensors()) { in Compile() local641 for (const auto& tensor_wrapper : op_wrapper->GetOutputTensors()) { in Compile() local
25 const std::shared_ptr<TensorWrapper>& tensor_wrapper, in RegisterIonMem()57 const std::shared_ptr<TensorWrapper>& tensor_wrapper, in RegisterCustomMem()150 const std::shared_ptr<TensorWrapper>& tensor_wrapper, in SetMemHandle()
94 const std::shared_ptr<TensorWrapper>& tensor_wrapper) { in EnsureTensorInQnnGraph()
20 for (const auto& tensor_wrapper : input_tensors_) { in GetOpConfig() local24 for (const auto& tensor_wrapper : output_tensors_) { in GetOpConfig() local
71 std::shared_ptr<TensorWrapper> tensor_wrapper = CreateTensorWrapper( variable
181 for (const auto& tensor_wrapper : op_wrapper->GetInputTensors()) { in Compile() local185 for (const auto& tensor_wrapper : op_wrapper->GetOutputTensors()) { in Compile() local
41 TF_TensorWrapper tensor_wrapper = TF_TensorWrapper(tf_tensor); in TestShapeMatch() local
262 TensorWrapper tensor_wrapper(tf.ones({4, 5})); in TEST_F() local
256 TF_Tensor* tensor_wrapper = in CreateTensor() local