Searched refs:AppendMajorDimension (Results 1 – 4 of 4) sorted by relevance
188 ShapeUtil::AppendMajorDimension(1, &index_shape); in TorchGather()267 ShapeUtil::AppendMajorDimension(1, &index_shape); in TorchIndexSelect()
365 static void AppendMajorDimension(int bound, Shape* shape);
315 /* static */ void ShapeUtil::AppendMajorDimension(int bound, Shape* shape) { in AppendMajorDimension() function in xla::ShapeUtil
521 ShapeUtil::AppendMajorDimension(1, &new_activation_shape); in HandleConvolution()