Searched refs:addressable_device_logical_ids (Results 1 – 5 of 5) sorted by relevance
47 addressable_device_logical_ids() const { in addressable_device_logical_ids() function48 return executable_->addressable_device_logical_ids(); in addressable_device_logical_ids()
343 auto span = exec->addressable_device_logical_ids(); in PYBIND11_MODULE()
607 std::vector<LogicalDeviceIds> addressable_device_logical_ids,637 absl::Span<const LogicalDeviceIds> addressable_device_logical_ids() in addressable_device_logical_ids() function
445 virtual absl::Span<const LogicalDeviceIds> addressable_device_logical_ids()
1570 std::vector<LogicalDeviceIds> addressable_device_logical_ids, in PjRtStreamExecutorExecutable() argument1577 std::move(addressable_device_logical_ids)), in PjRtStreamExecutorExecutable()2149 std::vector<PjRtExecutable::LogicalDeviceIds> addressable_device_logical_ids; in Compile() local2152 addressable_device_logical_ids.reserve(num_replicas * num_partitions); in Compile()2165 addressable_device_logical_ids.push_back(std::move(logica_device_ids)); in Compile()2187 std::move(device_assignment), std::move(addressable_device_logical_ids), in Compile()