- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ifrt_restore_variable (0.27 sec)
-
tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_mlrt_ops.td
); let results = (outs TFTensorType:$array_key, MlrtFutureType: $tensor_future ); } def IfrtRestoreVariableOp: TensorflowMlrt_Op<"ifrt_restore_variable", []> { let summary = "Restore variable tensors"; let description = [{ This is the MLRT version of tf.IfrtRestoreVariableOp. This Op is similar to a combination of RestoreV2 and AssignVariable Op, but
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir
// CHECK-NEXT: [[HANDLE:%.*]] = tf_mlrt.executeop %handle = "tf.VarHandleOp"() {__op_key = 3: i32, container = "x", shared_name = "y"} : () -> tensor<!tf_type.resource<tensor<3x1xf32>>> // CHECK-NEXT: "tf_mlrt.ifrt_restore_variable"([[PREFIX]], [[NAME]], [[SLICE]], [[HANDLE]]) <{restored_dtypes = [f32], truncate_in_cast = array<i1: true>}>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 24.7K bytes - Viewed (0)