Searched refs:AppendMajorDimension (Results 1 – 4 of 4) sorted by relevance
204 ShapeUtil::AppendMajorDimension(1, &index_shape); in TorchGather()286 ShapeUtil::AppendMajorDimension(1, &index_shape); in TorchIndexSelect()
300 static void AppendMajorDimension(int bound, Shape* shape);
459 /* static */ void ShapeUtil::AppendMajorDimension(int bound, Shape* shape) { in AppendMajorDimension() function in xla::ShapeUtil
532 ShapeUtil::AppendMajorDimension(1, &new_activation_shape); in HandleConvolution()