Searched refs:slice_shape_ptrs (Results 1 – 1 of 1) sorted by relevance
2221 std::vector<const Shape*> slice_shape_ptrs; in AllToAll() local2222 absl::c_transform(slice_shapes, std::back_inserter(slice_shape_ptrs), in AllToAll()2225 Shape shape, ShapeInference::InferAllToAllTupleShape(slice_shape_ptrs)); in AllToAll()