Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for StatelessCase (0.26 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/case_op.pbtxt

      attr {
        key: "value"
        value {
          tensor {
            dtype: DT_INT32
            tensor_shape {
            }
            int_val: 0
          }
        }
      }
    }
    node {
      name: "indexed_case"
      op: "StatelessCase"
      input: "Const_1"
      input: "Const"
      attr {
        key: "Tin"
        value {
          list {
            type: DT_INT32
          }
        }
      }
      attr {
        key: "Tout"
        value {
          list {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 07 20:38:34 UTC 2020
    - 3.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/case.mlir

          tf_executor.fetch %outputs_0 : tensor<*xi32>
        }
        func.return %0 : tensor<*xi32>
      }
    }
    
    // CHECK: name: "stateless_case"
    // CHECK-NEXT: "StatelessCase"
    // CHECK: name: "regular_case"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 28 12:06:33 UTC 2022
    - 2.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

        }
        ```
      }];
    
      let arguments = (ins
        I32Tensor:$branch_index,
        Variadic<TF_Tensor>:$input,
    
        ConfinedAttr<SymbolRefArrayAttr, [ArrayMinCount<1>]>:$branches,
    
        // Used to map StatelessCase and Case op defined in TensorFlow to a common
        // op.
        BoolAttr:$is_stateless
      );
    
      let results = (outs
        Variadic<TF_Tensor>:$output
      );
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/mark_for_compilation_pass.cc

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

        mlir::BoolAttr val = builder_.getBoolAttr(stateless);
        result.attributes.push_back(builder_.getNamedAttr("is_stateless", val));
      };
    
      // Map Case/If/While and StatelessCase/If/While op in TensorFlow to the common
      // Case/If/While op in MLIR and add the differentiating attribute.
      if (node.IsCaseNode()) composite_control_flow_op("Case");
      if (node.IsIfNode()) composite_control_flow_op("If");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  6. RELEASE.md

    *   Adds `tf.SparseTensor.with_values`. This returns a new SparseTensor with the
        same sparsity pattern, but with new provided values. It is similar to the
        `with_values` function of `RaggedTensor`.
    *   Adds `StatelessCase` op, and uses it if none of case branches has stateful
        ops.
    *   Adds `tf.config.experimental.get_memory_usage` to return total memory usage
        of the device.
    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