Home
last modified time | relevance | path

Searched defs:tensor_idx (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dobject_reader.cc38 uint32_t tensor_idx, Value** value) { in ReadNonConstantTensor()
112 absl::Status ObjectReader::ReadValueByTensorIdx(uint32_t tensor_idx, in ReadValueByTensorIdx()
142 const int tensor_idx = node_->inputs->data[idx]; in GetTensorDims() local
Dmodel_builder.cc2916 int tensor_idx = tensor_indices->data[i]; in IsAllAllowedTensors() local
3248 int tensor_idx = tflite_node->inputs->data[i]; in CopyVariableTensorOutputs() local
/external/tflite-support/tensorflow_lite_support/custom_ops/
Dtflite_inference_main.cc62 for (int tensor_idx : interpreter->inputs()) { in RunWithRandomInputs() local
87 for (int tensor_idx : interpreter->outputs()) { in RunWithRandomInputs() local
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights.cc74 int32_t tensor_idx) { in GetTensorConsumers()
192 int32_t tensor_idx = op->inputs[op_input_idx]; in CheckAllOpInputsQuantized() local
222 int32_t tensor_idx = op->inputs[op_input_idx]; in InsertQuantizableInputTensorsFromOperator() local
429 int32_t tensor_idx = tensor_pair.first; local
535 for (auto tensor_idx : op->inputs) { in QuantizeWeightsFloat16() local
559 int32_t tensor_idx = tensor_pair.first; in QuantizeWeightsFloat16() local
Dquantize_weights_portable.cc60 int32_t tensor_idx) { in GetTensorConsumers()
178 int32_t tensor_idx = op->inputs[op_input_idx]; in CheckAllOpInputsQuantized() local
208 int32_t tensor_idx = op->inputs[op_input_idx]; in InsertQuantizableInputTensorsFromOperator() local
415 int32_t tensor_idx = tensor_pair.first; local
521 for (auto tensor_idx : op->inputs) { in QuantizeWeightsFloat16() local
545 int32_t tensor_idx = tensor_pair.first; in QuantizeWeightsFloat16() local
Dquantize_model.cc134 bool IsFloatTensor(const SubGraphT* subgraph, int32_t tensor_idx) { in IsFloatTensor()
241 const int32_t tensor_idx = op->inputs[input_idx]; in PopulateRealValueOpSet() local
250 const int32_t tensor_idx = op->outputs[output_idx]; in PopulateRealValueOpSet() local
258 const int32_t tensor_idx = op->inputs[0]; in PopulateRealValueOpSet() local
265 const int32_t tensor_idx = op->outputs[0]; in PopulateRealValueOpSet() local
411 const int32_t tensor_idx, const TensorType& input_type, in SetInputType()
475 const int32_t tensor_idx, const TensorType& output_type, in SetOutputType()
891 const int32_t tensor_idx = op->inputs[input_idx]; in QuantizeOpInput() local
1619 for (int32_t tensor_idx = 0; tensor_idx < subgraph->tensors.size(); in GetAllOperatorOutputs() local
1662 const int32_t tensor_idx = op->inputs[input_idx]; in FillQuantizationParams() local
Dmodify_model_interface.cc405 const int32_t tensor_idx = subgraph->inputs[input_idx]; in AddUint8Dequant() local
445 const int32_t tensor_idx = subgraph->outputs[output_idx]; in AddUint8Quant() local
Dquantize_weights_test.cc97 bool IsModelInputOrOutput(const Model* model, uint32_t tensor_idx) { in IsModelInputOrOutput()
117 uint32_t tensor_idx, in GetProducerOpCode()
Dquantize_model_test.cc314 for (int tensor_idx = 0; tensor_idx < subgraph->tensors.size(); in TEST_P() local
392 for (int tensor_idx = 0; tensor_idx < subgraph->tensors.size(); in TEST_P() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator_test.cc107 for (int tensor_idx = 0; tensor_idx < 4; tensor_idx++) { in TEST() local
167 for (int tensor_idx = 0; tensor_idx < 7; tensor_idx++) { in TEST() local
186 for (int tensor_idx = 1; tensor_idx < 5; tensor_idx++) { in TEST() local
258 for (int tensor_idx = 0; tensor_idx < 7; tensor_idx++) { in TEST() local
277 for (int tensor_idx = 0; tensor_idx < 7; tensor_idx++) { in TEST() local
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dnativesignaturerunner_jni.cc65 for (int tensor_idx : signature_runner_->subgraph_->inputs()) { in GetInputIndex() local
78 for (int tensor_idx : signature_runner_->subgraph_->outputs()) { in GetOutputIndex() local
Dnativeinterpreterwrapper_jni.cc566 const int tensor_idx = interpreter->inputs()[input_idx]; in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput() local
/external/tensorflow/tensorflow/lite/python/analyzer_wrapper/
Dmodel_analyzer.cc69 const std::string tensor_str(const int tensor_idx, const int subgraph_idx, in tensor_str()
102 const tflite::Tensor* tensor, const int tensor_idx, in dump_tensor_detail()
157 const int tensor_idx = tensors->Get(i); in dump_tensor_list() local
/external/tensorflow/tensorflow/lite/testing/
Dselective_build_test.cc48 for (int tensor_idx : interpreter->inputs()) { in RunWithRandomInputs() local
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
Dquantize_weights_test.cc107 bool IsModelInputOrOutput(const Model* model, uint32_t tensor_idx) { in IsModelInputOrOutput()
127 uint32_t tensor_idx, BuiltinOperator* op_code) { in GetProducerOpCode()
/external/tensorflow/tensorflow/core/kernels/
Dsparse_conditional_accumulator.h75 const Tensor* tensor_idx = std::get<0>(*tensor); in ValidateShape() local
/external/ComputeLibrary/tests/validation/fixtures/
DGEMMFixture.h1179 int tensor_idx = 0; in compute_reference() local
1769 int tensor_idx = 0; in compute_reference() local
2284 int tensor_idx = 0; in compute_reference() local
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc177 const int tensor_idx) { in VerifyCustomAllocationForTensor()
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc1606 void AssignCustomAllocForTensor(int tensor_idx, int required_alignment) { in AssignCustomAllocForTensor()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc3029 auto is_const_tensor = [&node, &context](int tensor_idx) { in Validate()