Searched defs:tflite_tensor (Results 1 – 7 of 7) 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 | 81 const auto* tflite_tensor = ::tflite::GetInput(context_, node_, idx); in GetInput() local
97 auto* tflite_tensor = ::tflite::GetOutput(context_, node_, idx); in GetOutput() local
128 auto* tflite_tensor = ::tflite::GetInput(context_, node_, idx); in GetInputShape() local
150 const auto* tflite_tensor = ::tflite::GetInput(context_, node_, idx); in GetInputTensor() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_helper.cc | 96 absl::Status ExtractTensorShape(const TfLiteTensor& tflite_tensor, BHWC* bhwc) { in ExtractTensorShape()
122 absl::Status ExtractAxisFromIndex(const TfLiteTensor& tflite_tensor, int index, in ExtractAxisFromIndex()
156 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 | 41 TfLiteTensor tflite_tensor; in TEST() local
56 TfLiteTensor tflite_tensor; in TEST() local
72 TfLiteTensor tflite_tensor; in TEST() local
89 TfLiteTensor tflite_tensor; in TEST() local
107 TfLiteTensor tflite_tensor; in TEST() local
120 TfLiteTensor tflite_tensor; in TEST() local
|
| D | model_builder.cc | 2738 const TfLiteTensor& tflite_tensor = context->tensors[id]; in PrecreateIOTensors() local
|