Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 24 of 24 for use_node_name_sharing (0.33 sec)

  1. tensorflow/compiler/mlir/tfrt/tests/mlrt/parallelization.mlir

      %handle = "tf.HashTableV2"() {container = "", device = "/job:localhost/replica:0/task:0/device:CPU:0", key_dtype = !tf_type.string, shared_name = "hash_table", use_node_name_sharing = false, value_dtype = i64} : () -> tensor<!tf_type.resource>
    
    
      %keys = "tf.Const"() {value = dense<["a", "b", "c", "d"]> : tensor<4x!tf_type.string>} : () -> tensor<4x!tf_type.string>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 22:07:30 UTC 2023
    - 15K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfrt/tests/hoist_invariant_ops.mlir

      // CHECK: [[r:%.*]] = "tf.LookupTableFindV2"([[v]]#[[handle_idx]]
      // CHECK: return [[v]]#[[size_idx]], [[r]]
      %0 = "tf.HashTableV2"() {container = "", device = "", key_dtype = !tf_type.string, shared_name = "x", use_node_name_sharing = false, value_dtype = i64} : () -> tensor<!tf_type.resource>
      %1 = "tf.LookupTableSizeV2"(%0) {device = ""} : (tensor<!tf_type.resource>) -> tensor<i64>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 01 23:54:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir

            // expected-remark@above {{ID: 0}}
            %handle2 = "tf.HashTableV2"() {container = "", device = "", key_dtype = !tf_type.string, shared_name = "some_name", use_node_name_sharing = false, value_dtype = i64} : () -> tensor<!tf_type.resource>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Dec 20 04:39:18 UTC 2023
    - 129.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        DefaultValuedOptionalAttr<BoolAttr, "false">:$use_node_name_sharing,
        TypeAttr:$key_dtype,
        TypeAttr:$value_dtype
      );
    
      let results = (outs
        Res<TF_ResourceTensor, [{Handle to a table.}], [TF_LookupTableAlloc]>:$table_handle
      );
    
      let builders = [
        OpBuilder<(ins "StringAttr":$container, "StringAttr":$shared_name,
          "BoolAttr":$use_node_name_sharing, "TypeAttr":$key_dtype, "TypeAttr":$value_dtype),
    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