Searched refs:core_on_chip (Results 1 – 5 of 5) sorted by relevance
41 int core_on_chip() const { return core_.index(); } in core_on_chip() function
209 .def_property_readonly("core_on_chip", &TpuDevice::core_on_chip) in PYBIND11_MODULE()221 device.coords()[1], device.coords()[2], device.core_on_chip()); in PYBIND11_MODULE()
46 int core_on_chip);49 int core_on_chip() const { return core_on_chip_; } in core_on_chip() function
39 int core_on_chip) in TpuDevice() argument43 core_on_chip_(core_on_chip) {} in TpuDevice()
175 "core_on_chip", &PjRtTpuDevice::core_on_chip, in PYBIND11_MODULE()181 device.core_on_chip()); in PYBIND11_MODULE()