Searched defs:RegisterTypes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ | ||
| D | sharded_device_array.cc | 156 /*static*/ xla::Status ShardedDeviceArray::RegisterTypes(py::module& m) { in RegisterTypes() function in jax::ShardedDeviceArray |
| D | py_buffer.cc | 481 Status PyBuffer::RegisterTypes(py::module& m) { in RegisterTypes() function in xla::PyBuffer |