Searched defs:addressable_device_logical_ids (Results 1 – 3 of 3) sorted by relevance
47 addressable_device_logical_ids() const { in addressable_device_logical_ids() function
637 absl::Span<const LogicalDeviceIds> addressable_device_logical_ids() in addressable_device_logical_ids() function
1570 std::vector<LogicalDeviceIds> addressable_device_logical_ids, in PjRtStreamExecutorExecutable()2149 std::vector<PjRtExecutable::LogicalDeviceIds> addressable_device_logical_ids; in Compile() local