Home
last modified time | relevance | path

Searched refs:bias_enabled (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_ops.cc542 bool bias_enabled, bool pad_enabled, bool is_depthwise,
1186 bool fuse_biasadd_ = bias_enabled;
1550 bool pad_enabled, bool bias_enabled, bool is_depthwise,
1554 Tpadding, bias_enabled, false, is_depthwise,
1559 Tpadding, bias_enabled, false, is_depthwise, native_format>( in MklFusedDepthwiseConvOp()
1607 typename Ttemp_output, bool bias_enabled, bool is_depthwise,
1611 int32, bias_enabled, false, is_depthwise,
1628 bias_enabled, false, is_depthwise, native_format>(context) { in MklQuantizedConv2DOp()
1633 if (bias_enabled) { in MklQuantizedConv2DOp()
1645 bias_enabled, false, is_depthwise, in Compute()
[all …]
H A Dmkl_quantized_conv_ops_test.cc59 void RunQuantizedDepthwiseConv2DOp(const bool& bias_enabled) { in RunQuantizedDepthwiseConv2DOp() argument
82 if (bias_enabled) { in RunQuantizedDepthwiseConv2DOp()
113 if (bias_enabled) { in RunQuantizedDepthwiseConv2DOp()
H A Dmkl_conv_grad_filter_ops.cc353 template <typename Device, class T, bool bias_enabled, bool is_depthwise,
459 if (bias_enabled) { in Compute()
552 if (bias_enabled) { in Compute()
618 if (bias_enabled) { in Compute()
/aosp_15_r20/external/ComputeLibrary/src/graph/
H A DDataLayerVisitor.cpp66 std::string bias_enabled = node.input(2) == nullptr ? "0" : "1"; in add_convolution_layer_data() local
67 layer_data["bias_enabled"] = bias_enabled; in add_convolution_layer_data()
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit_autocast.py770 def __init__(self, bias_enabled=True): argument
772 self.conv = torch.nn.Conv2d(3, 64, 7, stride=2, bias=bias_enabled)
785 convbn(bias_enabled=True),
786 convbn(bias_enabled=False)]