- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for NCHW (0.03 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// Also FusedBatchNorm in training mode prefers NCHW data format. Check if all // users can efficiently use NHWC data format? if (one_by_one && trivial_strides && trivial_dilations) { return "NHWC"; } // If filter spatial dimensions are unknown or not 1x1 we prefer NCHW, because // it's the fastest option on NVIDIA GPUs with cuDNN library support. return "NCHW"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
// OK %0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "NCHW", ksize = [1, 1, 3, 6], padding = "VALID", strides = [1, 1, 3, 1]} : (tensor<1x16x6x6xf32>) -> tensor<1x16x1x1xf32> // Unsupported ksize %1 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "NCHW", ksize = [3, 1, 3, 6], padding = "VALID", strides = [1, 1, 3, 1]} : (tensor<1x16x6x6xf32>) -> tensor<1x16x1x1xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
} // ----- func.func @testBiasAdd(%arg0: tensor<2x3xf32>, %arg1: tensor<3xf32>) -> tensor<2x3xf32> { // expected-error @+1 {{requires value operand to have rank at least three with `NCHW` data format}} %0 = "tf.BiasAdd"(%arg0, %arg1) {data_format = "NCHW"} : (tensor<2x3xf32>, tensor<3xf32>) -> tensor<2x3xf32> func.return %0 : tensor<2x3xf32> } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
bool is_pool2d = rank == 4; if (verify_batch_channel_dims(0, rank - 1)) { *data_format = is_pool2d ? "NHWC" : "NDHWC"; } else if (verify_batch_channel_dims(0, 1)) { *data_format = is_pool2d ? "NCHW" : "NCDHW"; } else { return false; } if (rw.getWindowDilations().has_value() && !(rw.getWindowDilations()->isSplat() && rw.getWindowDilations()->getSplatValue<APInt>() == 1))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)