Searched refs:LogicalDeviceIds (Results 1 – 4 of 4) sorted by relevance
441 struct LogicalDeviceIds { struct445 virtual absl::Span<const LogicalDeviceIds> addressable_device_logical_ids() argument
607 std::vector<LogicalDeviceIds> addressable_device_logical_ids,637 absl::Span<const LogicalDeviceIds> addressable_device_logical_ids() in addressable_device_logical_ids()726 std::vector<LogicalDeviceIds> addressable_device_logical_ids_;
1570 std::vector<LogicalDeviceIds> addressable_device_logical_ids, in PjRtStreamExecutorExecutable()2149 std::vector<PjRtExecutable::LogicalDeviceIds> addressable_device_logical_ids; in Compile()2162 PjRtExecutable::LogicalDeviceIds logica_device_ids; in Compile()
46 absl::Span<const PjRtExecutable::LogicalDeviceIds>