Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for NCHW (0.06 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_lowering_patterns.td

          ConstBoolAttrFalse,
          ConstBoolAttrTrue),
        [(IsNhwcLayoutOp $attrs)]>;
    
    // pattern to lower a stablehlo.composite of `jax` and `pytorch` image resize fuctions
    // in `nearest`mode and with NCHW inputs to a tflite.resize_nearest_neighbor op.
    // TODO(b/343278954): Move the creation of transposes to a separate prepare pass
    // to avoid creating multiple pattern-rewrite rules for the same composite op.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 18:45:51 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  2. 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)
  3. tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td

    //===----------------------------------------------------------------------===//
    def IsDataFormatNHWC : ConstantAttr<TF_ConvnetDataFormatAttr, "\"NHWC\"">;
    def IsDataFormatNCHW : ConstantAttr<TF_ConvnetDataFormatAttr, "\"NCHW\"">;
    
    class I32VectorElementsAttr<int len> : ElementsAttrBase<
      CPred<"$_self.isa<DenseIntElementsAttr>() &&"
          "$_self.cast<DenseIntElementsAttr>().getType()."
          "getElementType().isSignlessInteger(32)">,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 28.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

    ```
    With `src_format` of `NHWC`, `dst_format` of `NCHW`, and input:
    ```
    [1, 2]
    ```
    , the output will be:
    ```
    [1, 2]
    ```
      }];
    
      let arguments = (ins
        Arg<TF_I32OrI64Tensor, [{Tensor of rank 1 or 2 in source data format.}]>:$x,
    
        DefaultValuedOptionalAttr<StrAttr, "\"NHWC\"">:$src_format,
        DefaultValuedOptionalAttr<StrAttr, "\"NCHW\"">:$dst_format
      );
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
  5. 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)
  6. 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)
Back to top