Searched refs:AppendMajorDimension (Results 1 – 4 of 4) sorted by relevance
407 ShapeUtil::AppendMajorDimension(group_size, &reshaped_activation_shape); in HandleConvolution()418 ShapeUtil::AppendMajorDimension(1, &reshaped_filter_shape); in HandleConvolution()424 ShapeUtil::AppendMajorDimension(1, &new_output_shape); in HandleConvolution()
141 ShapeUtil::AppendMajorDimension(1, &index_shape); in TorchGather()
355 static void AppendMajorDimension(int bound, Shape* shape);
268 /* static */ void ShapeUtil::AppendMajorDimension(int bound, Shape* shape) { in AppendMajorDimension() function in xla::ShapeUtil