Frank Chen 014f02fea5 Open source op definition for TPUReshardVariable op ... PiperOrigin-RevId: 358268125 Change-Id: I88c2de3f0bdb237cb1b235c2691a684ab644e916		2021-02-18 14:43:04 -08:00
..
BUILD	Open source op definition for TPUReshardVariable op	2021-02-18 14:43:04 -08:00
host_compute_ops.cc	Update outside compilation related passes to use tf._XlaHostComputeMlir and tf._TPUCompileMlirPlaceholderProgramKey instead of tf._HostComputeMlir and tf._TPUCompileMlir.	2020-07-29 13:25:45 -07:00
topk_ops.cc	Adjust TPU build dependencies.	2020-07-27 15:37:03 -07:00
tpu_compile_op.cc	[Resubmit] Only trigger resharding when necessary.	2020-09-16 16:23:31 -07:00
tpu_execute_op.cc	[Resubmit] Only trigger resharding when necessary.	2020-09-16 16:23:31 -07:00
tpu_partitioned_input_op.cc	Make sure TPUPartitionedInput shape inference doesn't crash if input handle shapes and types are not available.	2020-11-11 00:41:46 -08:00
tpu_partitioned_output_op.cc	Adjust TPU build dependencies.	2020-07-27 15:37:03 -07:00
tpu_reshard_variables_op.cc	Open source op definition for TPUReshardVariable op	2021-02-18 14:43:04 -08:00