Home
last modified time | relevance | path

Searched defs:sharded_token (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client_extension.cc229 py::class_<PyShardedTpuToken> sharded_token(m, "ShardedToken"); in PYBIND11_MODULE() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla.cc402 py::class_<PyShardedToken> sharded_token(m, "ShardedToken"); in PYBIND11_MODULE() local