Home
last modified time | relevance | path

Searched refs:bias_tensor_index (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_model_test.cc627 const int bias_tensor_index = 2; in TEST_F() local
630 subgraph->tensors[conv_op->inputs[bias_tensor_index]].get(); in TEST_F()
738 const int bias_tensor_index = 2; in TEST_P() local
741 subgraph->tensors[conv_op->inputs[bias_tensor_index]].get(); in TEST_P()
842 const int bias_tensor_index = 2; in TEST_P() local
845 subgraph->tensors[conv_op->inputs[bias_tensor_index]].get(); in TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_model_test.cc431 const int bias_tensor_index = 2; in TEST_P() local
434 subgraph->tensors[conv_op->inputs[bias_tensor_index]].get(); in TEST_P()
536 const int bias_tensor_index = 2; in TEST_P() local
539 subgraph->tensors[conv_op->inputs[bias_tensor_index]].get(); in TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dxnnpack_delegate.cc4627 const int bias_tensor_index = node->inputs->data[3]; in VisitTransposeConvNode() local
4628 if (bias_tensor_index != kTfLiteOptionalTensor) { in VisitTransposeConvNode()
4629 const TfLiteTensor& bias_tensor = tensors[bias_tensor_index]; in VisitTransposeConvNode()
4631 delegate, logging_context, bias_tensor, bias_tensor_index, in VisitTransposeConvNode()
4634 bias_tensor_index)); in VisitTransposeConvNode()
4635 if (quasi_static_tensors.count(bias_tensor_index) == 0) { in VisitTransposeConvNode()
4637 logging_context, bias_tensor, bias_tensor_index, node_index)); in VisitTransposeConvNode()
4640 xnnpack_tensor_bias = xnnpack_tensors[bias_tensor_index]; in VisitTransposeConvNode()