Searched refs:minor_dimension (Results 1 – 2 of 2) sorted by relevance
723 int64 minor_dimension = 0; member1093 ShapeUtil::GetDimension(this_shape, stride_config.minor_dimension); in PopulateInternal()1101 minor_scan_indexes[stride_config.minor_dimension] = i; in PopulateInternal()
95 minor_dimension = LayoutUtil::Minor(source_shape.layout(), 0); in StrideConfig()96 dest_stride = IndexUtil::GetDimensionStride(dest_shape, minor_dimension); in StrideConfig()98 minor_dimension = LayoutUtil::Minor(dest_shape.layout(), 0); in StrideConfig()100 IndexUtil::GetDimensionStride(source_shape, minor_dimension); in StrideConfig()102 minor_loop_size = dimensions[minor_dimension]; in StrideConfig()103 step[minor_dimension] = minor_loop_size; in StrideConfig()