Searched defs:tflite_tensor (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/shim/ |
| D | tflite_tensor_view_test.cc | 37 auto* tflite_tensor = interpreter.tensor(0); in TEST() local
59 auto* tflite_tensor = interpreter.tensor(0); in IntTest() local
87 auto* tflite_tensor = interpreter.tensor(0); in FloatTest() local
110 auto* tflite_tensor = interpreter.tensor(0); in TEST() local
153 auto* tflite_tensor = interpreter.tensor(0); in TEST() local
|
| D | tflite_op_shim.cc | 83 const auto* tflite_tensor = ::tflite::GetInput(context_, node_, idx); in GetInput() local
98 auto* tflite_tensor = ::tflite::GetOutput(context_, node_, idx); in GetOutput() local
132 auto* tflite_tensor = ::tflite::GetInput(context_, node_, idx); in GetInputShape() local
154 const auto* tflite_tensor = ::tflite::GetInput(context_, node_, idx); in GetInputTensor() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_helper.cc | 98 absl::Status ExtractTensorShape(const TfLiteTensor& tflite_tensor, BHWC* bhwc) { in ExtractTensorShape()
124 absl::Status ExtractAxisFromIndex(const TfLiteTensor& tflite_tensor, int index, in ExtractAxisFromIndex()
158 absl::Status ConvertTfLiteTensorToTensorRef(const TfLiteTensor& tflite_tensor, in ConvertTfLiteTensorToTensorRef()
|
| D | object_reader.cc | 45 TfLiteTensor* tflite_tensor = &context->tensors[tensor_idx]; in ReadNonConstantTensor() local
146 const TfLiteTensor& tflite_tensor = context_->tensors[tensor_idx]; in GetTensorDims() local
|
| D | object_reader.h | 77 const TfLiteTensor* tflite_tensor = context_->tensors + tensor_id; in ReadTensor() local
|
| D | model_builder_test.cc | 45 TfLiteTensor tflite_tensor; in TEST() local
60 TfLiteTensor tflite_tensor; in TEST() local
76 TfLiteTensor tflite_tensor; in TEST() local
93 TfLiteTensor tflite_tensor; in TEST() local
111 TfLiteTensor tflite_tensor; in TEST() local
124 TfLiteTensor tflite_tensor; in TEST() local
|
| D | model_builder.cc | 3232 const TfLiteTensor& tflite_tensor = context->tensors[id]; in PrecreateIOTensors() local
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util_test.cc | 196 TfLiteTensor tflite_tensor; in TEST() local
232 TfLiteTensor tflite_tensor; in TEST() local
|
| D | util.cc | 217 const TfLiteTensor* tflite_tensor) { in CreateTfTensorFromTfLiteTensor()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | delegate.cc | 182 const TfLiteTensor& tflite_tensor = context->tensors[tensor_index]; in Prepare() local
191 const TfLiteTensor& tflite_tensor = context->tensors[tensor_index]; in Prepare() local
|