History

Frank Chen 82f4f50f4f Add check for events_ not containing the event we are waiting on because it has already completed

PiperOrigin-RevId: 337185599
Change-Id: I9c73388c0a99c2abbc52aef4e7bf2c61656e8199

2020-10-14 15:39:09 -07:00

client

Remove the tensorflow/core:allocator alias. Use the real target in framework

2020-10-09 17:21:38 -07:00

platform/external

Remove the go_proto_library rules.

2020-10-06 16:42:56 -07:00

BUILD

Remove the go_proto_library rules.

2020-10-06 16:42:56 -07:00

direct_tpu_driver.cc

Add dependency for the direct TPU driver back into the tpu_driver target

2020-02-18 13:29:57 -08:00

event_id.h

TpuDriver: Improve handling of restarted clients/servers.

2019-11-15 14:04:48 -08:00

grpc_tpu_driver.cc

Cleanup TraceMe idioms

2020-07-31 23:39:13 -07:00

grpc_tpu_driver.h

Don't use grpc_impl:: as it is not API

2019-12-02 09:07:16 -08:00

pod_tpu_driver.cc

Add check for events_ not containing the event we are waiting on because it has already completed

2020-10-14 15:39:09 -07:00

README.md

…

recording_tpu_driver.cc

Adds optional flush functionality to recording_tpu_driver.cc

2020-10-07 11:51:45 -07:00

tpu_driver.cc

Add shape-based allocation and linearizer to external TPU driver

2020-01-09 11:43:21 -08:00

tpu_driver.h

Prefixing TensorFlow thread annotation macros with TF_.

2020-03-05 08:42:01 -08:00

tpu_driver.proto

Enable TPU POD for JAX/1VM by creating devices and local_devices topology.

2020-01-14 17:04:41 -08:00

tpu_service.proto

Implement GRPC TPU driver reset.

2019-11-20 17:49:06 -08:00

README.md

TPU Driver API

This repository contains the TPU driver API and network (gRPC) transport implementation for high-performance access to TPU hardware.

Building

Bazel is used to build the driver library and tests. Remote tests will require access to a Cloud TPU.

Fetching Bazel

Download the latest copy of Bazel from https://github.com/bazelbuild/bazel/releases.

Building

bazel build ...

Testing

bazel test ...