Searched defs:addressable_device_logical_ids (Results 1 – 5 of 5) sorted by relevance
48 addressable_device_logical_ids() const { in addressable_device_logical_ids() function
300 addressable_device_logical_ids; member710 absl::Span<const LogicalDeviceIds> addressable_device_logical_ids() in addressable_device_logical_ids() function
561 absl::Span<const LogicalDeviceIds> addressable_device_logical_ids() in addressable_device_logical_ids() function
1719 std::vector<LogicalDeviceIds> addressable_device_logical_ids, in PjRtStreamExecutorExecutable()2324 addressable_device_logical_ids = extras.addressable_device_logical_ids; in GetExecutableExtras() local2386 addressable_device_logical_ids = extras.addressable_device_logical_ids; in Compile() local
304 std::vector<PjRtExecutable::LogicalDeviceIds> addressable_device_logical_ids; in Compile() local1197 std::vector<LogicalDeviceIds> addressable_device_logical_ids, in TfrtCpuExecutable()