Home
last modified time | relevance | path

Searched refs:is_conv2d (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.h323 bool is_conv2d = (strides_.size() == 4); variable
325 if (is_conv2d) {
344 if (is_conv2d) {
355 if (is_conv2d) {
386 is_conv2d ? static_cast<int>(TF_2DFILTER_DIM_O)
394 if (is_conv2d) {
425 if (is_conv2d) {
446 is_conv2d
453 if (is_conv2d) {
Dmkl_conv_grad_filter_ops.cc376 bool is_conv2d = (this->strides_.size() == 4); in Compute() local
438 auto tf_fmt = is_conv2d in Compute()
461 : obp_tf_shape.dim_size(is_conv2d ? 3 : 4); in Compute()
487 if (is_conv2d) { in Compute()
Dmkl_conv_grad_input_ops.cc313 bool is_conv2d = (this->strides_.size() == 4); in Compute() local
377 auto tf_fmt = is_conv2d in Compute()
390 : (is_conv2d ? memory::format::hwio in Compute()
Dmkl_conv_ops.cc996 bool is_conv2d = (strides_.size() == 4); in Compute() local
998 if (!is_conv2d) { in Compute()
1006 OP_REQUIRES(context, is_conv2d, in Compute()
1012 if (!is_conv2d) { in Compute()
1020 auto tf_fmt = is_conv2d ? TFDataFormatToMklDnnDataFormat(data_format_) in Compute()
1039 auto filter_format = is_conv2d ? (is_depthwise ? memory::format::hwigo in Compute()