Searched refs:addressable_device_logical_ids_ (Results 1 – 2 of 2) sorted by relevance
1576 addressable_device_logical_ids_( in PjRtStreamExecutorExecutable()1973 const int replica = addressable_device_logical_ids_[0].replica; in Execute()1974 const int partition = addressable_device_logical_ids_[0].partition; in Execute()1984 const int replica = addressable_device_logical_ids_[i].replica; in Execute()1985 const int partition = addressable_device_logical_ids_[i].partition; in Execute()2036 const int replica = addressable_device_logical_ids_[i].replica; in Execute()2037 const int partition = addressable_device_logical_ids_[i].partition; in Execute()2069 argument_handles, addressable_device_logical_ids_[i].replica, in ExecuteSharded()2070 addressable_device_logical_ids_[i].partition, RunId(), options); in ExecuteSharded()
639 return addressable_device_logical_ids_; in addressable_device_logical_ids()726 std::vector<LogicalDeviceIds> addressable_device_logical_ids_; variable