Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for ResourceGather (0.21 sec)

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

      // CHECK: [[var:%.*]] = tf_mlrt.executeop.device([[device]]){{.*}}op: \22ResourceGather\22
      // CHECK: [[rendezvous_key_base:%.*]] = tf_mlrt_tpu.compile_and_execute([[var]])
      %0 = "tf.ResourceGather"(%arg0, %indices) {__op_key = 0: i32, device = "/device:CPU:0"} : (tensor<!tf_type.resource<tensor<4xf32>>>, tensor<i32>) -> tensor<f32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 04 21:25:31 UTC 2023
    - 11K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td

          ),
          (TF_AssignVariableOp $m_resource, $new_m, (CreateConstBoolAttrFalse)),
          (TF_AssignVariableOp $v_resource, $new_v, (CreateConstBoolAttrFalse))
        ]
      >;
    
    // Pattern to decompose tf.ResourceGather into tf.ReadVariable and tf.GatherV2.
    def DecomposeResourceGather : Pat<
      (TF_ResourceGatherOp:$old_result
        $resource, $indices, $batch_dims, $validate_indices),
      (TF_GatherV2Op:$dest
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 20.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/tf_tfl_passes.cc

      // 2) Convert composite functions like lstm/rnns, along with proper function
      // inlining & dce.
      // 3) Lower static tensor list pass.
    
      // This decomposes resource ops like ResourceGather into read-variable op
      // followed by gather. This is used when the saved model import path is used
      // during which resources don't get frozen in the python layer.
      pass_manager->addNestedPass<mlir::func::FuncOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 18:45:51 UTC 2024
    - 25.5K bytes
    - Viewed (0)
Back to top