Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dbias_op_base.py156 def bias_add(input_tensor, bias_tensor): argument
167 def bias_add_2(bias_tensor): argument
/external/ComputeLibrary/tests/validation/fixtures/
DQLSTMLayerNormalizationFixture.h61 void fill(InputType &&input_tensor, InputType &&weight_tensor, BiasType &&bias_tensor) in fill()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
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
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/mesa3d/src/gallium/targets/teflon/
Dtest_executor.cpp109 auto bias_tensor = subgraph->tensors[bias_index]; in patch_conv2d() local
285 auto bias_tensor = subgraph->tensors[bias_index]; in patch_fully_connected() local
/external/tensorflow/tensorflow/core/kernels/mkl/
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()
Dmkl_matmul_op_fused.cc65 const Tensor& bias_tensor = MklGetInput(ctx, this->kInputIndexBias); in Compute() local
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()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dmatmul_test.cc65 auto* bias_tensor = interpreter_->tensor(bias_); in FullyConnectedOpModel() local
Dtranspose_conv_test.cc95 auto* bias_tensor = interpreter_->tensor(bias_); in SetBias() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc211 const auto& bias_tensor = context->tensors[node->inputs->data[2]]; in IsNodeSupportedByHexagon() local
/external/mesa3d/src/gallium/frontends/teflon/
Dtfl_device.c513 TfLiteTensor *bias_tensor = &context->tensors[node->inputs->data[2]]; in PrepareDelegate() local
533 TfLiteTensor *bias_tensor = &context->tensors[node->inputs->data[2]]; in PrepareDelegate() local
/external/tensorflow/tensorflow/lite/tools/optimize/
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
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
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmkl_remapper_test.cc110 auto bias_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd() local
486 auto bias_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
Dremapper_test.cc2182 auto bias_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
/external/mesa3d/src/gallium/drivers/etnaviv/
Detnaviv_ml.h112 struct pipe_resource *bias_tensor; member
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
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
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
Dquantize_model_test.cc433 const auto bias_tensor = in TEST_P() local
538 const auto bias_tensor = in TEST_P() local
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc688 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
Dlstm.cc1128 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
/external/mesa3d/src/gallium/include/pipe/
Dp_state.h1100 struct pipe_tensor *bias_tensor; member
1187 struct pipe_tensor *bias_tensor; member
/external/pytorch/torch/csrc/jit/tensorexpr/
Dkernel.cpp615 auto bias_tensor = at::zeros({outputShape[1].AsNode<LongImm>()->value()}); in computeValue() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1079 DenseElementsAttr input_tensor, weights_tensor, bias_tensor; in fold() local

12