Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dbias_op_base.py156 def bias_add(input_tensor, bias_tensor): argument
167 def bias_add_2(bias_tensor): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dconv_2d_helpers.cc102 const TfLiteTensor& data_tensor, const TfLiteTensor& bias_tensor, in ProcessPerChannelQuantizedBias()
363 const auto& bias_tensor = context->tensors[inputs->data[2]]; in InitializeBiasNodes() local
H A Dtranspose_conv_2d_builder.cc141 const auto& bias_tensor = context->tensors[inputs->data[3]]; in PopulateSubGraph() local
H A Dmatmul_builder.cc93 const auto& bias_tensor = context->tensors[bias_tensor_id]; in AddFullyConnectedHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_qmatmul_op.cc165 const Tensor& bias_tensor = MklGetInput(context, this->kInputIndexBias); in Compute() local
405 const Tensor& bias_tensor, const Tensor& weight_tensor, in GetBiasHandle()
H A Dmkl_conv_ops.cc893 const Tensor& bias_tensor = MklGetInput(context, kInputIndex_Bias); in Compute() local
1076 const Tensor& bias_tensor) { in GetBiasHandle()
1771 const Tensor& bias_tensor) override { in GetBiasHandle()
1867 Tensor** bias_tensor) { in AllocateTensor()
H A Dmkl_matmul_op_fused.cc65 const Tensor& bias_tensor = MklGetInput(ctx, this->kInputIndexBias); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_model.cc87 const TensorT* bias_tensor = in DuplicateBiasesWithMultipleUses() local
276 const TensorT* weight_tensor, TensorT* bias_tensor, in QuantizeBias()
1572 TensorT* bias_tensor = subgraph->tensors[op->inputs[bias_idx]].get(); in QuantizeBiases() local
1777 TensorT* bias_tensor = subgraph->tensors[op->inputs[bias_idx]].get(); in EnsureBiasScaleCompatibility() local
H A Dquantize_model_test.cc629 const auto bias_tensor = in TEST_F() local
740 const auto bias_tensor = in TEST_P() local
844 const auto bias_tensor = in TEST_P() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DQLSTMLayerNormalizationFixture.h61 void fill(InputType &&input_tensor, InputType &&weight_tensor, BiasType &&bias_tensor) in fill()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmkl_remapper_test.cc110 auto bias_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd() local
486 auto bias_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
H A Dremapper_test.cc2182 auto bias_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/
H A Dutils.cc211 const auto& bias_tensor = context->tensors[node->inputs->data[2]]; in IsNodeSupportedByHexagon() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dmatmul_test.cc65 auto* bias_tensor = interpreter_->tensor(bias_); in FullyConnectedOpModel() local
H A Dtranspose_conv_test.cc95 auto* bias_tensor = interpreter_->tensor(bias_); in SetBias() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dxnnpack_delegate.cc2585 const TfLiteTensor& bias_tensor = tensors[bias_tensor_id]; in VisitConv2DNode() local
2694 const TfLiteTensor& bias_tensor = tensors[bias_tensor_id]; in VisitDepthwiseConv2DNode() local
2979 const TfLiteTensor& bias_tensor = tensors[bias_tensor_id]; in VisitFullyConnectedNode() local
3532 const TfLiteTensor& bias_tensor = tensors[node->inputs->data[2]]; in VisitMediaPipeDeconvolutionNode() local
4629 const TfLiteTensor& bias_tensor = tensors[bias_tensor_index]; in VisitTransposeConvNode() local
/aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/
H A Detnaviv_ml.h84 struct pipe_resource *bias_tensor; member
/aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/
H A Dtest_executor.cpp104 auto bias_tensor = subgraph->tensors[bias_index]; in patch_conv2d() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DQuantUtils.cpp65 const Shape& weight_shape, const int32_t* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_model_test.cc433 const auto bias_tensor = in TEST_P() local
538 const auto bias_tensor = in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunidirectional_sequence_lstm.cc688 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
H A Dlstm.cc1128 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.cpp615 auto bias_tensor = at::zeros({outputShape[1].AsNode<LongImm>()->value()}); in computeValue() local
/aosp_15_r20/external/mesa3d/src/gallium/include/pipe/
H A Dp_state.h1084 struct pipe_tensor *bias_tensor; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc1079 DenseElementsAttr input_tensor, weights_tensor, bias_tensor; in fold() local

12