Searched refs:compute_format (Results 1 – 1 of 1) sorted by relevance
796 TensorFormat compute_format = in operator ()() local808 << " compute format: " << ToString(compute_format); in operator ()()859 if (tensor_format == compute_format) { in operator ()()861 } else if (tensor_format == FORMAT_NHWC && compute_format == FORMAT_NCHW) { in operator ()()864 ShapeFromFormat(compute_format, batch_size, in operator ()()875 ShapeFromFormat(compute_format, batch_size, in operator ()()882 " and compute format: ", ToString(compute_format))); in operator ()()887 compute_format == FORMAT_NHWC ? se::dnn::DataLayout::kBatchYXDepth in operator ()()974 if (tensor_format == FORMAT_NHWC && compute_format == FORMAT_NCHW) { in operator ()()1012 TensorFormat compute_format = in operator ()() local[all …]