Searched refs:bias_enabled (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.cc | 415 bool bias_enabled, bool pad_enabled, bool is_depthwise,
949 bool fuse_biasadd_ = bias_enabled;
1285 bool pad_enabled, bool bias_enabled, bool is_depthwise,
1289 Tpadding, bias_enabled, false, is_depthwise,
1294 Tpadding, bias_enabled, false, is_depthwise, native_format>( in MklFusedDepthwiseConvOp()
1342 typename Ttemp_output, bool bias_enabled, bool is_depthwise>
1345 int32, bias_enabled, false, is_depthwise, false> {
1361 bias_enabled, false, is_depthwise, false>(context) { in MklQuantizedConv2DOp()
1366 if (bias_enabled) { in MklQuantizedConv2DOp()
1378 bias_enabled, false, is_depthwise, false>::Compute(context); in Compute()
[all …]
|
| D | mkl_quantized_conv_ops_test.cc | 98 void RunQuantizedDepthwiseConv2DOp(const bool& bias_enabled) { in RunQuantizedDepthwiseConv2DOp() argument
121 if (bias_enabled) { in RunQuantizedDepthwiseConv2DOp()
134 if (bias_enabled) { in RunQuantizedDepthwiseConv2DOp()
163 if (bias_enabled) { in RunQuantizedDepthwiseConv2DOp()
|
| D | mkl_conv_grad_filter_ops.cc | 359 template <typename Device, class T, bool bias_enabled, bool is_depthwise,
455 if (bias_enabled) { in Compute()
532 if (bias_enabled) { in Compute()
589 if (bias_enabled) { in Compute()
|