Searched refs:ResolveDeviceAssignment (Results 1 – 4 of 4) sorted by relevance
186 StatusOr<absl::optional<xla::DeviceAssignment>> ResolveDeviceAssignment(
223 StatusOr<absl::optional<xla::DeviceAssignment>> ResolveDeviceAssignment( in ResolveDeviceAssignment() function
79 ResolveDeviceAssignment(ctx, result->collective_reduce_info); in Run()
277 ResolveDeviceAssignment(ctx, compilation_result->collective_reduce_info); in Compute()