- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for IfrtRestoreVariableOp (0.36 sec)
-
tensorflow/compiler/mlir/tfrt/tests/ifrt/lower_to_ifrt_restore_variable.mlir
// CHECK-NEXT: [[HANDLEY:%.*]] = "tf.VarHandleOp"() <{container = "", shared_name = "y"}> : () -> tensor<!tf_type.resource<tensor<3x1xf32>>> // CHECK-NEXT: "tf.IfrtRestoreVariableOp"([[PREFIX]], [[NAME]], [[SLICE]], [[HANDLEY]]) // CHECK-SAME: {restored_dtypes = [f32], truncate_in_cast = array<i1: false>} // CHECK-NOT: "tf.RestoreV2" // CHECK-NEXT: return module {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 8.8K bytes - Viewed (0) -
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 this Op's execution is asynchronous.
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/tensorflow/ir/host_runtime/tfrt_ops.td
StrArrayAttr:$shared_name, StrArrayAttr:$container ); let results = (outs Variadic<TF_Tensor>:$results ); let hasVerifier = 1; } def TF_IfrtRestoreVariableOp : TF_Op<"IfrtRestoreVariableOp", []> { let summary = "Restore variable tensors"; let description = [{ This Op is similar to a combination of RestoreV2 and AssignVariable Op, but this Op's execution is asynchronous.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir
// CHECK-NEXT: mlrt.await_all_control // CHECK-NEXT: return func.return } // ----- // Test lowering of IfrtRestoreVariableOp // CHECK-LABEL: func @ifrt_restore_variable_test func.func @ifrt_restore_variable_test() -> () { // CHECK-NEXT: [[PREFIX:%.*]] = tf_mlrt.constop
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 24.7K bytes - Viewed (0)