Searched refs:src_tensor_indices (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | while.cc | 56 const SrcVector& src_tensor_indices, in CopyTensorsShapeAndType() argument
60 TF_LITE_ENSURE_EQ(context, src_tensor_indices.size(), in CopyTensorsShapeAndType()
62 for (int i = 0; i < src_tensor_indices.size(); ++i) { in CopyTensorsShapeAndType()
67 src_subgraph->tensor(src_tensor_indices[i]); in CopyTensorsShapeAndType()
88 const SrcVector& src_tensor_indices, in CopyTensorsData() argument
91 TF_LITE_ENSURE_EQ(context, src_tensor_indices.size(), in CopyTensorsData()
93 for (int i = 0; i < src_tensor_indices.size(); ++i) { in CopyTensorsData()
98 src_subgraph->tensor(src_tensor_indices[i]); in CopyTensorsData()
115 const SrcVector& src_tensor_indices, in DeepCopyTensorsShapeTypeData() argument
125 context, src_subgraph, src_tensor_indices, dst_subgraph, in DeepCopyTensorsShapeTypeData()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| H A D | call.cc | 148 const std::vector<int>& src_tensor_indices = src_subgraph->outputs(); in CopyOutputTensorsData() local
149 TF_LITE_ENSURE_EQ(context, src_tensor_indices.size(), node->outputs->size); in CopyOutputTensorsData()
150 for (int i = 0; i < src_tensor_indices.size(); ++i) { in CopyOutputTensorsData()
152 src_subgraph->tensor(src_tensor_indices[i]); in CopyOutputTensorsData()
|