Searched refs:logical_dim (Results 1 – 3 of 3) sorted by relevance
62 for (int logical_dim = 0; logical_dim < shape.rank(); logical_dim++) { in HandleReduce() local63 VLOG(5) << "Processing dimension " << logical_dim << " of size " in HandleReduce()64 << shape.dimensions(logical_dim); in HandleReduce()65 if (is_reduced(logical_dim) && logical_dim < shape.rank() - 1 && in HandleReduce()66 is_reduced(logical_dim + 1)) { in HandleReduce()69 next_dim_size *= shape.dimensions(logical_dim); in HandleReduce()73 if (is_reduced(logical_dim)) { in HandleReduce()75 shape.dimensions(logical_dim)); in HandleReduce()79 new_grouped_dims.push_back(shape.dimensions(logical_dim)); in HandleReduce()
73 int64 logical_dim = operand_layout.minor_to_major(major_to_minor_dim_idx); in HandleReduce() local74 int64 dim_size = operand_shape.dimensions(logical_dim); in HandleReduce()75 VLOG(5) << "Processing logical dimension " << logical_dim << " of size " in HandleReduce()79 if (absl::c_linear_search(reduce->dimensions(), logical_dim)) { in HandleReduce()83 int64 logical_reduce_dim = to_reduce_logical_dim(logical_dim); in HandleReduce()
792 enum isl_surf_dim logical_dim, in isl_surf_choose_dim_layout() argument802 switch (logical_dim) { in isl_surf_choose_dim_layout()827 switch (logical_dim) { in isl_surf_choose_dim_layout()