Searched refs:is_conv2d (Results 1 – 4 of 4) sorted by relevance
329 bool is_conv2d = (strides_.size() == 4); variable331 if (is_conv2d) {350 if (is_conv2d) {361 if (is_conv2d) {392 is_conv2d ? static_cast<int>(TF_2DFILTER_DIM_O)400 if (is_conv2d) {434 if (is_conv2d) {455 is_conv2d462 if (is_conv2d) {
431 bool is_conv2d = (this->strides_.size() == 4); in Compute() local433 auto tf_fmt = is_conv2d in Compute()459 : obp_tf_shape.dim_size(is_conv2d ? 3 : 4); in Compute()485 if (is_conv2d) { in Compute()
383 bool is_conv2d = (this->strides_.size() == 4); in Compute() local388 auto tf_fmt = is_conv2d in Compute()405 : (is_conv2d ? memory::format_tag::hwio in Compute()
575 bool is_conv2d = (strides_.size() == 4); in Compute() local577 if (!is_conv2d) { in Compute()588 OP_REQUIRES(context, is_conv2d, in Compute()596 auto tf_fmt = is_conv2d ? TFDataFormatToMklDnnDataFormat(data_format_) in Compute()620 auto filter_format = is_conv2d ? (is_depthwise ? memory::format_tag::hwigo in Compute()