Searched refs:slice_shape_ptrs (Results 1 – 1 of 1) sorted by relevance
2956 std::vector<const Shape*> slice_shape_ptrs; in AllToAllTuple() local2957 absl::c_transform(slice_shapes, std::back_inserter(slice_shape_ptrs), in AllToAllTuple()2960 Shape shape, ShapeInference::InferAllToAllTupleShape(slice_shape_ptrs)); in AllToAllTuple()