Home
last modified time | relevance | path

Searched refs:bias_name (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Densure_bias_vectors.cc65 const std::string& bias_name = in ProcessLinearOperator() local
67 op->inputs.push_back(bias_name); in ProcessLinearOperator()
68 auto& bias_array = model->GetOrCreateArray(bias_name); in ProcessLinearOperator()
Dfuse_binary_into_preceding_affine.cc118 const auto& bias_name = preceding_op->inputs[bias_ind]; in FuseMulOrDivParamsIntoPrecedingAffine() local
121 auto& bias = model->GetArray(bias_name); in FuseMulOrDivParamsIntoPrecedingAffine()
122 DropMinMax(model, bias_name); in FuseMulOrDivParamsIntoPrecedingAffine()
305 const auto& bias_name = preceding_op->inputs[bias_ind]; in Run() local
307 const auto& bias = model->GetArray(bias_name); in Run()
320 const int count_ops_consuming_bias = CountOpsWithInput(*model, bias_name); in Run()
Dfuse_binary_into_following_affine.cc124 const auto& bias_name = following_op->inputs[2]; in FuseMulOrDivParamsIntoFollowingAffine() local
127 DropMinMax(model, bias_name); in FuseMulOrDivParamsIntoFollowingAffine()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_metal.cc533 std::string bias_name = "args.biases.GetPtr()"; in GenerateConvolution() local
536 bias_name = "bias_loc"; in GenerateConvolution()
543 c += " r" + r_id + " += TO_ACCUM_TYPE(" + bias_name + "[" + in GenerateConvolution()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc530 const std::string& bias_name = in ConvertDepthwiseConvOperator() local
532 const auto& bias_array = model.GetArray(bias_name); in ConvertDepthwiseConvOperator()
539 ConvertFloatTensorConst(bias_name, bias_shape_1d, bias_data, in ConvertDepthwiseConvOperator()
1537 const std::string bias_name = WalkUpToConstantArray( in ConvertLstmCellOperator() local
1539 const auto& bias_array = model.GetArray(bias_name); in ConvertLstmCellOperator()
Dimport_tensorflow.cc1041 const auto& bias_name = node.input(1); in ConvertBiasAddOperator() local
1045 biasadd->inputs.push_back(bias_name); in ConvertBiasAddOperator()
/external/tensorflow/tensorflow/lite/python/
Dlite_test.py1144 bias_name = 'Conv2D' if enable_mlir_converter else 'Conv2D_bias'
1154 bias_name)
1187 bias_name)