Home
last modified time | relevance | path

Searched defs:tensor_index (Results 1 – 25 of 46) sorted by relevance

12

/external/tensorflow/tensorflow/lite/
Darena_planner.cc118 for (int tensor_index : graph_info_->outputs()) { in PlanAllocations() local
124 for (int tensor_index : graph_info_->variables()) { in PlanAllocations() local
137 for (int tensor_index : graph_info_->inputs()) { in PlanAllocations() local
151 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
165 int tensor_index = node_outputs->data[j]; in PlanAllocations() local
174 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
204 int tensor_index = node_temporaries->data[j]; in ExecuteAllocations() local
309 for (const auto& tensor_index : tensor_order) { in CalculateAllocations() local
318 for (const auto& tensor_index : tensor_order) { in CalculateAllocations() local
339 TfLiteStatus ArenaPlanner::ResolveTensorAllocation(int tensor_index) { in ResolveTensorAllocation()
Dinterpreter.cc164 int tensor_index, const TfLiteCustomAllocation& allocation) { in SetCustomAllocationForTensor()
264 TfLiteStatus Interpreter::ResizeInputTensor(int tensor_index, in ResizeInputTensor()
270 int tensor_index, const std::vector<int>& dims) { in ResizeInputTensorStrict()
288 for (int tensor_index : outputs()) { in Invoke() local
308 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadOnly()
317 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadWrite()
326 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadOnly()
336 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadWrite()
415 TfLiteStatus Interpreter::SetBufferHandle(int tensor_index, in SetBufferHandle()
434 TfLiteStatus Interpreter::GetBufferHandle(int tensor_index, in GetBufferHandle()
Dinterpreter.h250 TfLiteTensor* tensor(int tensor_index) { in tensor()
255 const TfLiteTensor* tensor(int tensor_index) const { in tensor()
269 T* typed_tensor(int tensor_index) { in typed_tensor()
281 const T* typed_tensor(int tensor_index) const { in typed_tensor()
337 const int tensor_index = GetTensorIndexFromSignatureDefName( in input_tensor_by_signature_name() local
349 const int tensor_index = GetTensorIndexFromSignatureDefName( in output_tensor_by_signature_name() local
523 TfLiteStatus EnsureTensorDataIsReadable(int tensor_index) { in EnsureTensorDataIsReadable()
Doptional_debug_tools.cc115 for (size_t tensor_index = 0; tensor_index < interpreter->tensors_size(); in PrintInterpreterState() local
Darena_planner_test.cc215 std::ptrdiff_t GetOffset(int tensor_index) { in GetOffset()
222 std::ptrdiff_t GetOffsetAfter(int tensor_index) { in GetOffsetAfter()
233 bool IsUnallocated(int tensor_index) { in IsUnallocated()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.h51 void SetTypedValues(int tensor_index, const std::vector<T>& values) { in SetTypedValues()
58 std::vector<T> GetTypedValues(int tensor_index) { in GetTypedValues()
65 void SetValues(int tensor_index, const std::vector<float>& values) { in SetValues()
71 std::vector<float> GetValues(int tensor_index) { in GetValues()
Dtest_util.cc28 void FlexModelTest::SetStringValues(int tensor_index, in SetStringValues()
51 void FlexModelTest::SetShape(int tensor_index, const std::vector<int>& values) { in SetShape()
56 std::vector<int> FlexModelTest::GetShape(int tensor_index) { in GetShape()
66 TfLiteType FlexModelTest::GetType(int tensor_index) { in GetType()
70 bool FlexModelTest::IsDynamicTensor(int tensor_index) { in IsDynamicTensor()
95 void FlexModelTest::SetConstTensor(int tensor_index, in SetConstTensor()
Dkernel.cc387 for (auto tensor_index : TfLiteIntArrayView(params->output_tensors)) { in Init() local
393 for (auto tensor_index : TfLiteIntArrayView(params->input_tensors)) { in Init() local
456 for (auto tensor_index : op_data_->subgraph_inputs) { in Prepare() local
482 for (auto tensor_index : op_data_->subgraph_outputs) { in Prepare() local
599 for (auto tensor_index : op_data_->subgraph_inputs) { in Eval() local
621 for (auto tensor_index : op_data_->subgraph_outputs) { in Eval() local
Dbuffer_map.cc157 void BufferMap::SetFromTfLite(int tensor_index, const TfLiteTensor* tensor) { in SetFromTfLite()
181 void BufferMap::SetFromTensorFlow(int tensor_index, tensorflow::Tensor tensor) { in SetFromTensorFlow()
/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.cc37 int tensor_index) { in GetTensorAtIndex()
49 int* tensor_index) { in ValidateTensorIndexingSafe()
72 const int tensor_index = tensor_indices[index]; in ValidateTensorIndexing() local
82 const int tensor_index = ValidateTensorIndexing( in GetMutableInput() local
93 int tensor_index; in GetMutableInputSafe() local
121 const int tensor_index = ValidateTensorIndexing( in GetOutput() local
131 int tensor_index; in GetOutputSafe() local
147 const int tensor_index = ValidateTensorIndexing( in GetTemporary() local
158 int tensor_index; in GetTemporarySafe() local
168 const int tensor_index = ValidateTensorIndexing( in GetIntermediates() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc98 const auto tensor_index = node->inputs->data[input_idx]; in Eval() local
124 for (auto tensor_index : TfLiteIntArrayView(node->outputs)) { in Eval() local
211 for (auto tensor_index : TfLiteIntArrayView(node->inputs)) { in Prepare() local
214 for (auto tensor_index : TfLiteIntArrayView(node->outputs)) { in Prepare() local
217 for (auto tensor_index : tensors) { in Prepare() local
/external/tensorflow/tensorflow/lite/micro/
Dmicro_interpreter.h114 T* typed_tensor(int tensor_index) { in typed_tensor()
130 T* typed_input_tensor(int tensor_index) { in typed_input_tensor()
146 T* typed_output_tensor(int tensor_index) { in typed_output_tensor()
Dmicro_allocator.cc215 const int tensor_index = subgraph->inputs()->Get(i); in AddTensors() local
222 const int tensor_index = subgraph->outputs()->Get(i); in AddTensors() local
231 const int tensor_index = op->inputs()->Get(n); in AddTensors() local
238 const int tensor_index = op->outputs()->Get(n); in AddTensors() local
867 const Model* model, TfLiteEvalTensor* eval_tensors, int tensor_index) { in AllocatePersistentTfLiteTensor()
899 const Model* model, TfLiteEvalTensor* eval_tensors, int tensor_index) { in AllocateTempTfLiteTensor()
992 const Model* model, TfLiteEvalTensor* eval_tensors, int tensor_index) { in AllocatePersistentTfLiteTensorInternal()
999 int tensor_index, bool allocate_temp) { in PopulateTfLiteTensorFromFlatbuffer()
Drecording_micro_allocator.cc201 const Model* model, TfLiteEvalTensor* eval_tensors, int tensor_index) { in AllocatePersistentTfLiteTensorInternal()
213 int tensor_index, bool allocate_temp) { in PopulateTfLiteTensorFromFlatbuffer()
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dgl_delegate.cc80 int tensor_index; member
122 absl::Status BindBufferToTensor(GLuint ssbo, int tensor_index) { in BindBufferToTensor()
149 auto find_value = [&](int tensor_index) -> Value* { in Prepare()
173 const int tensor_index = delegate_params->input_tensors->data[i]; in Prepare() local
215 const int tensor_index = delegate_params->output_tensors->data[i]; in Prepare() local
483 int tensor_index) { in TfLiteGpuDelegateBindBufferToTensor()
/external/tensorflow/tensorflow/lite/python/
Dinterpreter.py390 def _get_tensor_details(self, tensor_index): argument
483 def set_tensor(self, tensor_index, value): argument
682 def get_tensor(self, tensor_index): argument
697 def tensor(self, tensor_index): argument
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib.cc170 for (auto tensor_index : in ExportTensors() local
177 for (int tensor_index = 0; tensor_index < subgraph_->tensors_size(); in ExportTensors() local
186 for (int tensor_index = 0; tensor_index < subgraph_->tensors_size(); in ExportTensors() local
339 for (int tensor_index : TfLiteIntArrayView(node.inputs)) { in CheckInputOutput() local
368 for (int tensor_index : outputs) { in CheckInputOutput() local
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dgpu_api_delegate.cc197 for (auto tensor_index : input_refs) { in Prepare() local
204 for (auto tensor_index : output_refs) { in Prepare() local
233 void BindGlBufferToTensor(GLuint buffer_id, int tensor_index, in BindGlBufferToTensor()
382 TfLiteDelegate* delegate, GLuint buffer_id, int tensor_index, in TfLiteGpuDelegateBindGlBufferToTensor()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dkernel_runner.cc77 int tensor_index) { in GetTensor()
86 const struct TfLiteContext* context, int tensor_index) { in GetEvalTensor()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibration_logger.h52 TfLiteStatus LogTensorValue(int tensor_index, const float* tensor_values, in LogTensorValue()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h147 TfLiteTensor* tensor(int tensor_index) { in tensor()
156 const TfLiteTensor* tensor(int tensor_index) const { in tensor()
275 TfLiteStatus EnsureTensorDataIsReadable(int tensor_index) { in EnsureTensorDataIsReadable()
Dsubgraph.cc421 for (int tensor_index : node_subset.output_tensors) { in ReplaceNodeSubsetsWithDelegateKernels() local
808 TfLiteStatus Subgraph::ResizeInputTensor(int tensor_index, in ResizeInputTensor()
842 TfLiteStatus Subgraph::ResizeInputTensorStrict(int tensor_index, in ResizeInputTensorStrict()
1037 int tensor_index = node.inputs->data[i]; in Invoke() local
1194 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadOnly()
1253 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadWrite()
1594 int tensor_index, const TfLiteCustomAllocation& allocation) { in SetCustomAllocationForTensor()
/external/tensorflow/tensorflow/core/protobuf/tpu/
Dtpu_embedding_output_layout.proto26 int32 tensor_index = 1; field
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h71 inline QuantizedType GetIntermediateElementType(LstmOp op, int tensor_index) { in GetIntermediateElementType()
530 for (int tensor_index : in GetLstmOpQuantSpec() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc626 int tensor_index, int node_index) { in CheckTensorType()
638 int tensor_index, int node_index) { in CheckTensorFloatType()
646 int tensor_index) { in CheckTensorShape()
689 int tensor_index) { in CheckTensorShape()
696 int tensor_index, int node_index) { in CheckSlopeTensorShape()
723 int tensor_index, in CheckPaddingsTensorShape()
755 int tensor_index, int node_index) { in CheckAxesTensorShape()
769 int tensor_index, int node_index) { in CheckShapeTensorShape()
782 TfLiteContext* context, const TfLiteTensor& tensor, int tensor_index, in CheckTensorNonDynamicAllocation()
798 int tensor_index, in CheckTensorStaticAllocation()

12