| /external/tensorflow/tensorflow/lite/ |
| D | arena_planner.cc | 107 for (int tensor_index : graph_info_->outputs()) { in PlanAllocations() local
113 for (int tensor_index : graph_info_->variables()) { in PlanAllocations() local
119 for (int tensor_index : graph_info_->inputs()) { in PlanAllocations() local
129 for (int tensor_index : graph_info_->variables()) { in PlanAllocations() local
142 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
150 for (int tensor_index : graph_info_->inputs()) { in PlanAllocations() local
162 int tensor_index = node_outputs->data[j]; in PlanAllocations() local
171 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
247 TfLiteStatus ArenaPlanner::ResolveTensorAllocation(int tensor_index) { in ResolveTensorAllocation()
264 TfLiteStatus ArenaPlanner::CalculateTensorAllocation(int tensor_index) { in CalculateTensorAllocation()
[all …]
|
| D | interpreter.h | 252 TfLiteTensor* tensor(int tensor_index) { in tensor()
257 const TfLiteTensor* tensor(int tensor_index) const { in tensor()
270 T* typed_tensor(int tensor_index) { in typed_tensor()
282 const T* typed_tensor(int tensor_index) const { in typed_tensor()
381 TfLiteStatus EnsureTensorDataIsReadable(int tensor_index) { in EnsureTensorDataIsReadable()
|
| D | interpreter.cc | 128 TfLiteStatus Interpreter::ResizeInputTensor(int tensor_index, in ResizeInputTensor()
137 for (int tensor_index : outputs()) { in Invoke() local
156 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadOnly()
165 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadWrite()
174 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadOnly()
188 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadWrite()
241 TfLiteStatus Interpreter::SetBufferHandle(int tensor_index, in SetBufferHandle()
261 TfLiteStatus Interpreter::GetBufferHandle(int tensor_index, in GetBufferHandle()
|
| D | optional_debug_tools.cc | 88 for (size_t tensor_index = 0; tensor_index < interpreter->tensors_size(); in PrintInterpreterState() local
|
| D | arena_planner_test.cc | 185 int64_t GetOffset(int tensor_index) { in GetOffset()
192 int64_t GetOffsetAfter(int tensor_index) { in GetOffsetAfter()
423 auto is_unallocated = [&](int tensor_index) { in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | test_util.h | 49 void SetTypedValues(int tensor_index, const std::vector<T>& values) { in SetTypedValues()
56 std::vector<T> GetTypedValues(int tensor_index) { in GetTypedValues()
63 void SetValues(int tensor_index, const std::vector<float>& values) { in SetValues()
69 std::vector<float> GetValues(int tensor_index) { in GetValues()
|
| D | test_util.cc | 28 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()
|
| D | kernel.cc | 385 for (auto tensor_index : TfLiteIntArrayView(params->output_tensors)) { in Init() local
391 for (auto tensor_index : TfLiteIntArrayView(params->input_tensors)) { in Init() local
465 for (auto tensor_index : op_data->subgraph_inputs) { in Prepare() local
481 for (auto tensor_index : op_data->subgraph_outputs) { in Prepare() local
518 for (auto tensor_index : op_data->subgraph_inputs) { in Eval() local
540 for (auto tensor_index : op_data->subgraph_outputs) { in Eval() local
|
| D | buffer_map.cc | 147 void BufferMap::SetFromTfLite(int tensor_index, const TfLiteTensor* tensor) { in SetFromTfLite()
171 void BufferMap::SetFromTensorFlow(int tensor_index, tensorflow::Tensor tensor) { in SetFromTensorFlow()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | interpreter.py | 115 def _get_tensor_details(self, tensor_index): argument
173 def set_tensor(self, tensor_index, value): argument
216 def get_tensor(self, tensor_index): argument
231 def tensor(self, tensor_index): argument
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 126 TfLiteTensor* tensor(int tensor_index) { in tensor()
135 const TfLiteTensor* tensor(int tensor_index) const { in tensor()
248 TfLiteStatus EnsureTensorDataIsReadable(int tensor_index) { in EnsureTensorDataIsReadable()
|
| D | subgraph.cc | 306 for (int tensor_index : node_subset.output_tensors) { in ReplaceNodeSubsetsWithDelegateKernels() local
580 TfLiteStatus Subgraph::ResizeInputTensor(int tensor_index, in ResizeInputTensor()
705 int tensor_index = node.inputs->data[i]; in Invoke() local
820 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadOnly()
871 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadWrite()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/ |
| D | micro_interpreter.cc | 87 const int tensor_index = subgraph_->inputs()->Get(i); in MicroInterpreter() local
109 const int tensor_index = op->inputs()->Get(n); in MicroInterpreter() local
115 const int tensor_index = op->outputs()->Get(n); in MicroInterpreter() local
|
| /external/tensorflow/tensorflow/lite/experimental/writer/ |
| D | writer_lib.cc | 131 for (auto tensor_index : in ExportTensors() local
138 for (int tensor_index = 0; tensor_index < interpreter_->tensors_size(); in ExportTensors() local
147 for (int tensor_index = 0; tensor_index < interpreter_->tensors_size(); in ExportTensors() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibration_logger.h | 67 void LogTensorValue(int tensor_index, const float* tensor_values, in LogTensorValue()
|
| D | calibrator.cc | 199 for (auto tensor_index : tensor_indices) { in GetLoggableTensorIndices() local
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | tpu_embedding_output_layout.proto | 22 int32 tensor_index = 1; field
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | subgraph_test_util.cc | 46 void SetupTensor(Subgraph* subgraph, int tensor_index, TfLiteType type) { in SetupTensor()
350 int tensor_index, in CreateConstantInt32Tensor()
|
| D | if.cc | 156 for (int tensor_index : active_branch_subgraph.outputs()) { in Eval() local
|
| D | while.cc | 268 for (int tensor_index : body_subgraph->outputs()) { in Eval() local
|
| /external/tensorflow/tensorflow/lite/profiling/ |
| D | profile_summarizer.cc | 33 int tensor_index) { in GetTensorName()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 274 TfLiteStatus AddTensorInput(int tensor_index, bool hybrid_op) { in AddTensorInput()
278 TfLiteStatus AddTensorOutput(int tensor_index) { in AddTensorOutput()
287 TfLiteStatus AddStateFloat32Tensor(int tensor_index, in AddStateFloat32Tensor()
404 TfLiteStatus AddTensor(int tensor_index, bool hybrid_op, in AddTensor()
|
| /external/tensorflow/tensorflow/lite/java/src/main/native/ |
| D | tensor_jni.cc | 34 TensorHandle(tflite::Interpreter* interpreter, int tensor_index) in TensorHandle()
298 JNIEnv* env, jclass clazz, jlong interpreter_handle, jint tensor_index) { in Java_org_tensorflow_lite_Tensor_create()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.cc | 302 PyObject* CheckGetTensorArgs(Interpreter* interpreter_, int tensor_index, in CheckGetTensorArgs()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array_ops.cc | 424 const Tensor* tensor_index; in Compute() local
509 const Tensor* tensor_index; in Compute() local
|