Searched refs:is_depthwise (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 67 const std::vector<int32>& dilations, bool is_depthwise = false)
195 bool is_depthwise) { in GetFilterSizeInMklOrder() argument
233 if (is_depthwise) { in GetFilterSizeInMklOrder()
289 bool is_depthwise) { in GetFilterSizeInMklOrder() argument
293 is_depthwise); in GetFilterSizeInMklOrder()
323 bool is_depthwise = false) {
387 if (is_depthwise) {
489 memory::dims* pad_r, bool is_depthwise) { in GetOutputAndPadSizeInMklOrder() argument
513 is_depthwise); in GetOutputAndPadSizeInMklOrder()
530 bool is_depthwise = false) {
[all …]
|
| D | mkl_conv_ops.cc | 415 bool bias_enabled, bool pad_enabled, bool is_depthwise,
543 &padding_right, (fuse_pad_ || pad_attr_enabled), is_depthwise); in Compute()
587 if (is_depthwise) { in Compute()
620 auto filter_format = is_conv2d ? (is_depthwise ? memory::format_tag::hwigo in Compute()
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()
[all …]
|
| D | mkl_conv_grad_input_ops.cc | 303 template <typename Device, class T, bool is_depthwise, bool native_format>
305 : public MklConvBackpropCommonOp<Device, T, is_depthwise> {
308 : MklConvBackpropCommonOp<Device, T, is_depthwise>(context) {} in MklConvCustomBackpropInputOp()
380 &padding_left, &padding_right, false, is_depthwise); in Compute()
403 is_depthwise in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 359 template <typename Device, class T, bool bias_enabled, bool is_depthwise,
362 : public MklConvBackpropCommonOp<Device, T, is_depthwise> {
365 : MklConvBackpropCommonOp<Device, T, is_depthwise>(context) {} in MklConvCustomBackpropFilterOp()
428 &padding_left, &padding_right, false, is_depthwise); in Compute()
486 if (!is_depthwise) { in Compute()
668 return is_depthwise in GetOutputFormat()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 122 bool is_depthwise) { in CanFuseConvOrDepthwiseConvShapes() argument
146 (is_depthwise ? filter_shape.back() != elements_depth in CanFuseConvOrDepthwiseConvShapes()
153 bool is_depthwise) { in CanFuseConvOrDepthwiseConv() argument
161 is_depthwise); in CanFuseConvOrDepthwiseConv()
165 bool is_depthwise) { in CanFuseConvOrDepthwiseConv() argument
170 is_depthwise); in CanFuseConvOrDepthwiseConv()
203 ElementsAttr ExpandTo4DForConvImpl(Attribute a, bool is_depthwise) { in ExpandTo4DForConvImpl() argument
212 if (is_depthwise) in ExpandTo4DForConvImpl()
|
| D | optimize_patterns.td | 91 class CanFuseConvOrDepthwiseConv<string is_depthwise> : Constraint<
92 CPred<"TFL::CanFuseConvOrDepthwiseConv($0, $1, " # is_depthwise # ")">>;
|
| /external/XNNPACK/bench/ |
| D | convolution.cc | 529 bool is_depthwise = false; in tflite_convolution_f32() local
532 is_depthwise = true; in tflite_convolution_f32()
568 is_depthwise ? tflite::BuiltinOperator_DEPTHWISE_CONV_2D : tflite::BuiltinOperator_CONV_2D, in tflite_convolution_f32()
648 …is_depthwise ? tflite::BuiltinOptions_DepthwiseConv2DOptions : tflite::BuiltinOptions_Conv2DOption… in tflite_convolution_f32()
649 is_depthwise ? dwconv2d_options.Union() : conv2d_options.Union(), in tflite_convolution_f32()
935 bool is_depthwise = false; in armcl_convolution_f32() local
940 is_depthwise = true; in armcl_convolution_f32()
948 if (is_depthwise) { in armcl_convolution_f32()
|