| /external/tflite-support/tensorflow_lite_support/custom_ops/ |
| D | tflite_inference_main.cc | 62 for (int tensor_idx : interpreter->inputs()) { in RunWithRandomInputs() local
87 for (int tensor_idx : interpreter->outputs()) { in RunWithRandomInputs() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | object_reader.cc | 38 uint32_t tensor_idx, Value** value) { in ReadNonConstantTensor()
112 absl::Status ObjectReader::ReadValueByTensorIdx(uint32_t tensor_idx, in ReadValueByTensorIdx()
129 const int tensor_idx = node_->inputs->data[idx]; in GetTensorDims() local
|
| D | object_reader.h | 66 const int32_t tensor_idx = node_->inputs->data[idx]; in ReadTensor() local
|
| D | model_builder.cc | 2458 int tensor_idx = tensor_indices->data[i]; in IsAllAllowedTensors() local
2565 int tensor_idx = tflite_node->inputs->data[i]; in CopyVariableTensorOutputs() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibrator_test.cc | 105 for (int tensor_idx = 0; tensor_idx < 4; tensor_idx++) { in TEST() local
164 for (int tensor_idx = 0; tensor_idx < 7; tensor_idx++) { in TEST() local
181 for (int tensor_idx = 1; tensor_idx < 5; tensor_idx++) { in TEST() local
251 for (int tensor_idx = 0; tensor_idx < 7; tensor_idx++) { in TEST() local
270 for (int tensor_idx = 0; tensor_idx < 7; tensor_idx++) { in TEST() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_weights.cc | 60 int32_t tensor_idx) { in GetTensorConsumers()
177 int32_t tensor_idx = op->inputs[op_input_idx]; in CheckAllOpInputsQuantized() local
207 int32_t tensor_idx = op->inputs[op_input_idx]; in InsertQuantizableInputTensorsFromOperator() local
456 int32_t tensor_idx = tensor_pair.first; in QuantizeWeightsInt8() local
561 for (auto tensor_idx : op->inputs) { in QuantizeWeightsFloat16() local
585 int32_t tensor_idx = tensor_pair.first; in QuantizeWeightsFloat16() local
|
| D | quantize_model.cc | 42 bool IsFloatTensor(const SubGraphT* subgraph, int32_t tensor_idx) { in IsFloatTensor()
107 const int32_t tensor_idx = op->inputs[input_idx]; in PopulateRealValueOpSet() local
116 const int32_t tensor_idx = op->outputs[output_idx]; in PopulateRealValueOpSet() local
124 const int32_t tensor_idx = op->inputs[0]; in PopulateRealValueOpSet() local
131 const int32_t tensor_idx = op->outputs[0]; in PopulateRealValueOpSet() local
260 const int32_t tensor_idx, const TensorType& input_type, in SetInputType()
324 const int32_t tensor_idx, const TensorType& output_type, in SetOutputType()
615 const int32_t tensor_idx = op->inputs[input_idx]; in QuantizeOpInput() local
1157 for (int32_t tensor_idx = 0; tensor_idx < subgraph->tensors.size(); in GetAllOperatorOutputs() local
1192 const int32_t tensor_idx = op->inputs[input_idx]; in FillQuantizationParams() local
|
| D | modify_model_interface.cc | 402 const int32_t tensor_idx = subgraph->inputs[input_idx]; in AddUint8Dequant() local
442 const int32_t tensor_idx = subgraph->outputs[output_idx]; in AddUint8Quant() local
|
| D | quantize_weights_test.cc | 97 bool IsModelInputOrOutput(const Model* model, uint32_t tensor_idx) { in IsModelInputOrOutput()
117 uint32_t tensor_idx, in GetProducerOpCode()
|
| D | quantize_model_test.cc | 303 for (int tensor_idx = 0; tensor_idx < subgraph->tensors.size(); in TEST_P() local
381 for (int tensor_idx = 0; tensor_idx < subgraph->tensors.size(); in TEST_P() local
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | selective_build_test.cc | 48 for (int tensor_idx : interpreter->inputs()) { in RunWithRandomInputs() local
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_interpreter.cc | 84 int tensor_idx) { in GetTensor()
91 const struct TfLiteContext* context, int tensor_idx) { in GetEvalTensor()
|
| D | micro_allocator_test.cc | 85 TfLiteEvalTensor* eval_tensors, int tensor_idx, in VerifyMockTensor()
94 TfLiteEvalTensor* eval_tensors, int tensor_idx) { in VerifyMockWeightTensor()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_conditional_accumulator.h | 87 const Tensor* tensor_idx = std::get<0>(*tensor); in ValidateShape() local
|
| /external/tensorflow/tensorflow/lite/java/src/main/native/ |
| D | nativeinterpreterwrapper_jni.cc | 642 const int tensor_idx = interpreter->inputs()[input_idx]; in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput() local
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 1534 void AssignCustomAllocForTensor(int tensor_idx, int required_alignment) { in AssignCustomAllocForTensor()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 2367 auto is_const_tensor = [&node, &context](int tensor_idx) { in Validate()
|