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