Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for stack728 (0.13 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

        maximum size needs to be a constant and 2) a push op can be found with a
        known shape, and all push ops need to have the same shape.
    
        A stack creation op "tf.StackV2" will be turned in to two zero-initialized
        variables, for the buffer and current size. Each push will be turned into
        ```mlir
          %old_val = "tf.ReadVariableOp"(%buffer)
          %old_size = "tf.ReadVariableOp"(%size)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      let results = (outs
        Res<TF_Tensor, [{The same tensor as the input 'elem'.}]>:$output
      );
    
      TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<1>;
    }
    
    def TF_StackV2Op : TF_Op<"StackV2", [TF_UniqueResourceAllocation]> {
      let summary = "A stack that produces elements in first-in last-out order.";
    
      let arguments = (ins
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top