Searched refs:parse_sharding_from_function (Results 1 – 1 of 1) sorted by relevance
1846 std::function<Status(const Edge* call_edge)> parse_sharding_from_function = in InferXlaShardingFromNeighbors() local1873 TF_RETURN_IF_ERROR(parse_sharding_from_function(out_edge)); in InferXlaShardingFromNeighbors()1879 TF_RETURN_IF_ERROR(parse_sharding_from_function(edge)); in InferXlaShardingFromNeighbors()