Home
last modified time | relevance | path

Searched refs:with_bias (Results 1 – 8 of 8) sorted by relevance

/external/ComputeLibrary/tests/validation/fixtures/
DFuseBatchNormalizationFixture.h51 …hape_w, DataType data_type, DataLayout data_layout, bool in_place, bool with_bias, bool with_gamma… in setup() argument
53 …et_b) = compute_target(shape_w, data_type, data_layout, in_place, with_bias, with_gamma, wit… in setup()
54 …std::tie(_reference_w, _reference_b) = compute_reference(shape_w, data_type, with_bias, with_gamma… in setup()
64 …hape_w, DataType data_type, DataLayout data_layout, bool in_place, bool with_bias, bool with_gamma… in compute_target() argument
74 const bool in_place_b = with_bias ? in_place : false; in compute_target()
86 auto b_to_use = with_bias ? &b : nullptr; in compute_target()
141 …sor<T>> compute_reference(TensorShape shape_w, DataType data_type, bool with_bias, bool with_gamma… in compute_reference() argument
162 if(!with_bias) in compute_reference()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_test.cc112 ops::BiasAdd with_bias = ops::BiasAdd( in RunMatMulWithBias() local
131 ops::BiasAdd with_bias = ops::BiasAdd( in RunMatMulWithBiasAndActivation() local
136 ops::Relu(root.WithOpName("with_activation"), with_bias); in RunMatMulWithBiasAndActivation()
138 ops::Relu6(root.WithOpName("with_activation"), with_bias); in RunMatMulWithBiasAndActivation()
140 ops::Elu(root.WithOpName("with_activation"), with_bias); in RunMatMulWithBiasAndActivation()
142 ops::internal::LeakyRelu(root.WithOpName("with_activation"), with_bias); in RunMatMulWithBiasAndActivation()
144 ops::Identity(root.WithOpName("with_activation"), with_bias); in RunMatMulWithBiasAndActivation()
Dconv_ops_test.cc569 ops::BiasAdd with_bias = ops::BiasAdd( in RunConv2DWithBias() local
590 ops::BiasAdd with_bias = ops::BiasAdd( in RunConv2DWithBiasAndActivation() local
595 ops::Relu(root.WithOpName("with_activation"), with_bias); in RunConv2DWithBiasAndActivation()
597 ops::Relu6(root.WithOpName("with_activation"), with_bias); in RunConv2DWithBiasAndActivation()
599 ops::Elu(root.WithOpName("with_activation"), with_bias); in RunConv2DWithBiasAndActivation()
601 ops::internal::LeakyRelu(root.WithOpName("with_activation"), with_bias); in RunConv2DWithBiasAndActivation()
603 ops::Identity(root.WithOpName("with_activation"), with_bias); in RunConv2DWithBiasAndActivation()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_matmul_acl.h41 bool with_bias; member
Druntime_conv2d_acl.h50 bool with_bias; member
Druntime_conv2d_acl.cc59 acl_conf.with_bias = false; in ACLDepthwiseConvImpl()
162 acl_conf.with_bias = false; in ACLGemmConvImpl()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Dquantization_ops_test.cc131 const std::string& data_format = "NCHW", bool with_bias = true, argument
151 if (with_bias) {
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_ops_test.cc1351 auto with_bias = ops::BiasAdd( in RunMklPadWithFusedConv2DAndBias() local
1379 auto with_bias = ops::BiasAdd( in RunMklPadWithFusedConv2DAndBiasRelu() local
1383 auto with_relu = ops::Relu(root.WithOpName("with_relu"), with_bias); in RunMklPadWithFusedConv2DAndBiasRelu()