Search Options

Results per page
Sort
Preferred Languages
Advance

Results 81 - 83 of 83 for ArrayAttr (0.4 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

    def TF_InfeedDequeueTupleOp : TF_Op<"InfeedDequeueTuple", []> {
      let summary = "Fetches multiple values from infeed as an XLA tuple.";
    
      let arguments = (ins
        OptionalAttr<StrAttr>:$_XlaSharding,
        OptionalAttr<ArrayAttr>:$layouts
      );
    
      let results = (outs
        Variadic<TF_Tensor>:$outputs
      );
    
      TF_DerivedResultShapeListAttr shapes = TF_DerivedResultShapeListAttr<0>;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

    // Return true if the op has non-empty "minmax" attribute.
    static inline bool HasValidMinMaxAttribute(Operation* op) {
      auto minmax = op->getAttrOfType<ArrayAttr>("minmax");
      return minmax && minmax.getValue().size() == 2;
    }
    
    namespace {
    
    /// This pattern matches and remove a tfl.fake_quant if all the users of this op
    /// and itself have "minmax" attribute set.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir

        %0 = "tf.B"(%arg0) : (tensor<?xi32>) -> tensor<?xi32>
        func.return %0 : tensor<?xi32>
      }
    }
    
    // -----
    
    // Tests Functions referenced by TPU function via SymbolRefAttr nested in
    // ArrayAttr and DictionaryAttr.
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 172.9K bytes
    - Viewed (0)
Back to top