Searched defs:is_depthwise (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 192 bool is_depthwise) { in GetFilterSizeInMklOrder()
294 bool is_depthwise) { in GetFilterSizeInMklOrder()
543 memory::dims* pad_r, bool is_grouped_convolution, bool is_depthwise) { in GetOutputAndPadSizeInMklOrder()
|
| D | mkl_conv_ops.cc | 2214 accu_type, has_bias, is_depthwise, is_native) \ argument
2226 is_depthwise, is_native) \ argument
2234 is_depthwise, is_native) \ argument
2241 op, kernel, output_type, accu_type, has_bias, is_depthwise, is_native) \ argument
2251 accu_type, has_bias, is_depthwise, is_native) argument
2300 accu_type, has_bias, is_depthwise, is_native) \ argument
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 142 bool is_depthwise) { in CanFuseConvOrDepthwiseConvShapes()
168 bool is_depthwise) { in CanFuseConvOrDepthwiseConv()
180 bool is_depthwise) { in CanFuseConvOrDepthwiseConv()
267 ElementsAttr ExpandTo4DForConvImpl(Attribute a, bool is_depthwise) { in ExpandTo4DForConvImpl()
|
| /external/XNNPACK/bench/ |
| D | convolution.cc | 522 bool is_depthwise = false; in tflite_convolution_f32() local
|