Searched refs:inner_dim_idx (Results 1 – 7 of 7) sorted by relevance
248 const int inner_dim_idx = IsColMajor ? 0 : NumDims - 1;260 output_offsets[inner_dim_idx] = offset;307 static_cast<Index>(m_impl.dimensions()[inner_dim_idx]);315 const Index output_inner_dim_size = desc.dimension(inner_dim_idx);320 input_offsets[inner_dim_idx] < 0321 ? numext::mini(numext::abs(input_offsets[inner_dim_idx]),330 numext::maxi(input_inner_dim_size - (input_offsets[inner_dim_idx] +370 (input_inner_dim_size == m_dimensions[inner_dim_idx]) &&374 const int squeeze_dim = IsColMajor ? inner_dim_idx + 1 : inner_dim_idx - 1;
270 static const Index inner_dim_idx = isColMajor ? 0 : NumDims - 1;271 const bool inner_dim_reversed = m_reverse[inner_dim_idx];
1479 const int inner_dim_idx = is_col_major ? 0 : NumDims - 1;1480 IndexType output_inner_dim_size = target.dims[inner_dim_idx];1483 eigen_assert(target.strides[inner_dim_idx] == 1);
254 static const Index inner_dim_idx = isColMajor ? 0 : NumDims - 1;255 const bool inner_dim_reversed = m_reverse[inner_dim_idx];
3780 int inner_dim_idx = lhs_layout.blocks().front().dim_idx; in build_tile_stmt() local3781 bool do_broadcast = po.is_broadcast_dim(inner_dim_idx); in build_tile_stmt()3789 ir_assert(lhs0.dim_idx == inner_dim_idx); in build_tile_stmt()3800 inner_tile_dims[inner_dim_idx] = lhs0.block; in build_tile_stmt()