Searched refs:get_sharding_for_node (Results 1 – 1 of 1) sorted by relevance
80 auto get_sharding_for_node = [](const Node* n) -> xla::StatusOr<int> { in ComputeArgAndRetvalCores() local96 TF_ASSIGN_OR_RETURN(int core, get_sharding_for_node(n)); in ComputeArgAndRetvalCores()103 TF_ASSIGN_OR_RETURN(int core, get_sharding_for_node(n)); in ComputeArgAndRetvalCores()108 TF_ASSIGN_OR_RETURN(retval_cores[index], get_sharding_for_node(n)); in ComputeArgAndRetvalCores()