Home
last modified time | relevance | path

Searched refs:minor_dimension (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h723 int64 minor_dimension = 0; member
1093 ShapeUtil::GetDimension(this_shape, stride_config.minor_dimension); in PopulateInternal()
1101 minor_scan_indexes[stride_config.minor_dimension] = i; in PopulateInternal()
Dliteral.cc95 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()