Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFDataFormatIsNDHWC (0.39 sec)

  1. tensorflow/compiler/mlir/lite/utils/validators.h

    }
    
    // Returns true if the given TensorFlow op does not have a `data_format`
    // attribute (then default to "NDHWC"), or its `data_format` attribute is
    // "NDHWC".
    inline bool TFDataFormatIsNDHWC(Operation *op) {
      auto attr = op->getAttrOfType<StringAttr>("data_format");
      return !attr || attr.getValue() == "NDHWC";
    }
    
    // Returns true if the given `op`
    //   * has an attribute with the given `name`,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 4.7K bytes
    - Viewed (0)
Back to top