Home
last modified time | relevance | path

Searched defs:with_bias (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DFuseBatchNormalizationFixture.h51 … shape_w, DataType data_type, DataLayout data_layout, bool in_place, bool with_bias, bool with_gam… in setup()
64 … shape_w, DataType data_type, DataLayout data_layout, bool in_place, bool with_bias, bool with_gam… in compute_target()
141 …ensor<T>> compute_reference(TensorShape shape_w, DataType data_type, bool with_bias, bool with_gam… in compute_reference()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_op_test.cc112 ops::BiasAdd with_bias = ops::BiasAdd( in RunMatMulWithBias() local
131 ops::BiasAdd with_bias = ops::BiasAdd( in RunMatMulWithBiasAndActivation() local
H A Dconv_ops_test.cc569 ops::BiasAdd with_bias = ops::BiasAdd( in RunConv2DWithBias() local
590 ops::BiasAdd with_bias = ops::BiasAdd( in RunConv2DWithBiasAndActivation() local
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dquantization.py2294 with_bias=False, argument
2336 with_bias=False, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_matmul_acl.h41 bool with_bias; member
H A Druntime_conv2d_acl.h50 bool with_bias; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/
H A DMatmul.cpp58 bool with_bias = false; in matmul() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear.cpp840 bool with_bias = bias_.has_value(); in apply_impl() local
993 bool with_bias = bias.has_value(); in linear_int8_with_onednn_weight() local
H A Dqconv.cpp1193 bool with_bias = bias_.has_value(); in apply_impl() local
1556 bool with_bias = bias.has_value(); in _quantized_convolution_onednn() local
H A Dqlinear_dynamic.cpp565 bool with_bias = bias_.has_value(); in apply_dynamic_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Dquantization_ops_test.cc131 const std::string& data_format = "NCHW", bool with_bias = true,
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_ops_test.cc1351 auto with_bias = ops::BiasAdd( in RunMklPadWithFusedConv2DAndBias() local
1379 auto with_bias = ops::BiasAdd( in RunMklPadWithFusedConv2DAndBiasRelu() local