- Sort Score
- Result 10 results
- Languages All
Results 31 - 34 of 34 for NCHW (0.15 sec)
-
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) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
// CHECK-SAME: %[[VAL_0:.*]]: tensor<4x3x16x16xf32>) -> tensor<4x3x7x7xf32> { // CHECK: %[[VAL_1:.*]] = "tf.AvgPool"(%[[VAL_0]]) <{data_format = "NCHW", ksize = [1, 1, 3, 3], padding = "VALID", strides = [1, 1, 2, 2]}> : (tensor<4x3x16x16xf32>) -> tensor<4x3x7x7xf32> // CHECK: return %[[VAL_1]] : tensor<4x3x7x7xf32> // CHECK: }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 340.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// is handled, which can optionally require a general broadcast of the // 'bias' term in a way that is not compatible with the standard left-padded // broadcast semantics (i.e. NCHW will broadcast into dimension 1). // The correct 'bias' broadcast will be synthesized manually. class ConvertBiasAddOp : public OpRewritePattern<TF::BiasAddOp> { public: using OpRewritePattern::OpRewritePattern;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
RELEASE.md
* Add Mel-Frequency Cepstral Coefficient support to `tf.contrib.signal` (with GPU and gradient support). * Add a self-check on `import tensorflow` for Windows DLL issues. * Add NCHW support to `tf.depth_to_space` on GPU. * TensorFlow Debugger (tfdbg): * Add `eval` command to allow evaluation of arbitrary Python/numpy expressions in tfdbg command-line interface. See
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0)