Define IfrtRestoreVariable op and its mlrt kernel for TFRT + IFRT stack (0152a415) · Commits · gitlab-org / build / omnibus-mirror / tensorflow · JiHu GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

T

tensorflow
- Activity
- Members
- Labels
- Issues
  0
- Issue boards
- Milestones
- Iterations
- Wiki
- Requirements
- Environments
- Terraform modules
- Incidents
- Service Desk

Snippets Groups Projects

gitlab-org
build
omnibus-mirror
tensorflow
Commits
0152a415

Commit 0152a415 authored 1 year ago by

Deqiang Chen Committed by TensorFlower Gardener 1 year ago

Downloads
- Patches
- Plain Diff

Define IfrtRestoreVariable op and its mlrt kernel for TFRT + IFRT stack

PiperOrigin-RevId: 613697238

parent dae44bc0

Changes 11

Hide whitespace changes

Inline Side-by-side

Showing

tensorflow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.td 30 additions, 0 deletions

...flow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.td
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc 1 addition, 1 deletion

...ler/mlir/tf2xla/transforms/legalization_op_config_test.cc
tensorflow/core/tfrt/ifrt/BUILD 19 additions, 0 deletions

tensorflow/core/tfrt/ifrt/BUILD
tensorflow/core/tfrt/ifrt/ifrt_model_context.h 18 additions, 0 deletions

tensorflow/core/tfrt/ifrt/ifrt_model_context.h
tensorflow/core/tfrt/ifrt/ifrt_restore_tensor_registry.cc 59 additions, 0 deletions

tensorflow/core/tfrt/ifrt/ifrt_restore_tensor_registry.cc
tensorflow/core/tfrt/ifrt/ifrt_restore_tensor_registry.h 55 additions, 0 deletions

tensorflow/core/tfrt/ifrt/ifrt_restore_tensor_registry.h
tensorflow/core/tfrt/mlrt/kernel/BUILD 10 additions, 0 deletions

tensorflow/core/tfrt/mlrt/kernel/BUILD
tensorflow/core/tfrt/mlrt/kernel/ifrt_ops_kernel.cc 166 additions, 0 deletions

tensorflow/core/tfrt/mlrt/kernel/ifrt_ops_kernel.cc
tensorflow/core/tfrt/mlrt/kernel/ifrt_ops_kernel_test.cc 232 additions, 0 deletions

tensorflow/core/tfrt/mlrt/kernel/ifrt_ops_kernel_test.cc
tensorflow/core/tfrt/mlrt/kernel/testdata/BUILD 52 additions, 0 deletions

tensorflow/core/tfrt/mlrt/kernel/testdata/BUILD
tensorflow/core/tfrt/mlrt/kernel/testdata/gen_checkpoint.py 68 additions, 0 deletions

tensorflow/core/tfrt/mlrt/kernel/testdata/gen_checkpoint.py

with 710 additions and 1 deletion

0% or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment