- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ResourceGather (0.42 sec)
-
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) -
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) -
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)