Home
last modified time | relevance | path

Searched defs:bias_tensor (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_helpers.cc89 const auto& bias_tensor = context->tensors[inputs->data[2]]; in ProcessPerChannelQuantizedBias() local
255 const auto& bias_tensor = context->tensors[inputs->data[2]]; in InitializeBiasNodes() local
Dtranspose_conv_2d_builder.cc141 const auto& bias_tensor = context->tensors[inputs->data[3]]; in PopulateSubGraph() local
Dmatmul_builder.cc93 const auto& bias_tensor = context->tensors[bias_tensor_id]; in AddFullyConnectedHelper() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dbias_op_base.py160 def bias_add(input_tensor, bias_tensor): argument
171 def bias_add_2(bias_tensor): argument
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_qmatmul_op.cc165 const Tensor& bias_tensor = MklGetInput(context, this->kInputIndexBias); in Compute() local
397 const Tensor& bias_tensor, const Tensor& weight_tensor, in GetBiasHandle()
Dmkl_matmul_op_fused.cc61 const Tensor& bias_tensor = MklGetInput(ctx, this->kInputIndexBias); in Compute() local
Dmkl_conv_ops.cc729 const Tensor& bias_tensor = MklGetInput(context, kInputIndex_Bias); in Compute() local
840 const Tensor& bias_tensor) { in GetBiasHandle()
1503 const Tensor& bias_tensor) override { in GetBiasHandle()
1600 Tensor** bias_tensor) { in AllocatePersistentTensor()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmkl_remapper_test.cc110 auto bias_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd() local
480 auto bias_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsvdf.cc58 const TfLiteEvalTensor* bias_tensor, in EvalIntegerSvdfHifimini()
254 const TfLiteEvalTensor* bias_tensor, const TfLiteSVDFParams* params, in EvalIntegerSvdfHifi4()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dsvdf_common.cc36 const TfLiteEvalTensor* bias_tensor, in EvalIntegerSvdfReference()
Dconv_test.cc482 TfLiteTensor bias_tensor = in TF_LITE_MICRO_TEST() local
558 TfLiteTensor bias_tensor = in TF_LITE_MICRO_TEST() local
695 TfLiteTensor bias_tensor = in TF_LITE_MICRO_TEST() local
Ddepthwise_conv_test.cc191 TfLiteTensor bias_tensor = CreatePerChannelQuantizedBiasTensor( in TestDepthwiseConvQuantizedPerChannel() local
732 TfLiteTensor bias_tensor = in TF_LITE_MICRO_TEST() local
819 TfLiteTensor bias_tensor = in TF_LITE_MICRO_TEST() local
966 TfLiteTensor bias_tensor = in TF_LITE_MICRO_TEST() local
Dconv_test_common.cc209 TfLiteTensor bias_tensor = CreatePerChannelQuantizedBiasTensor( in TestConvQuantizedPerChannel() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc211 const auto& bias_tensor = context->tensors[node->inputs->data[2]]; in IsNodeSupportedByHexagon() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dtranspose_conv_test.cc93 auto* bias_tensor = interpreter_->tensor(bias_); in SetBias() local
Dmatmul_test.cc66 auto* bias_tensor = interpreter_->tensor(bias_); in FullyConnectedOpModel() local
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc142 const TensorT* weight_tensor, TensorT* bias_tensor, in QuantizeBias()
1110 TensorT* bias_tensor = subgraph->tensors[op->inputs[bias_idx]].get(); in QuantizeBiases() local
1303 TensorT* bias_tensor = subgraph->tensors[op->inputs[bias_idx]].get(); in EnsureBiasScaleCompatibility() local
Dquantize_model_test.cc614 const auto bias_tensor = in TEST_F() local
723 const auto bias_tensor = in TEST_P() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc1259 const TfLiteTensor& bias_tensor = tensors[bias_tensor_id]; in VisitConv2DNode() local
1354 const TfLiteTensor& bias_tensor = tensors[bias_tensor_id]; in VisitDepthwiseConv2DNode() local
1587 const TfLiteTensor& bias_tensor = tensors[bias_tensor_id]; in VisitFullyConnectedNode() local
2044 const TfLiteTensor& bias_tensor = tensors[node->inputs->data[2]]; in VisitMediaPipeDeconvolutionNode() local
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dsvdf.cc210 const TfLiteEvalTensor* bias_tensor, in EvalIntegerSVDF()
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dconv_slicing_test.cc232 TfLiteTensor bias_tensor = CreatePerChannelQuantizedBiasTensor( in TestConvQuantizedPerChannel() local
Ddepthwise_conv_slicing_test.cc147 TfLiteTensor bias_tensor = CreatePerChannelQuantizedBiasTensor( in TestDepthwiseConvQuantizedPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc680 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
Dlstm.cc1127 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
/external/XNNPACK/bench/
Dconvolution.cc895 arm_compute::Tensor bias_tensor; in armcl_convolution_f32() local

12