Searched refs:GetDimensionStride (Results 1 – 3 of 3) sorted by relevance
70 static int64 GetDimensionStride(const Shape& shape, int64 dimension);
134 /* static */ int64 IndexUtil::GetDimensionStride(const Shape& shape, in GetDimensionStride() function in xla::IndexUtil
75 dest_stride = IndexUtil::GetDimensionStride(dest_shape, minor_dimension); in StrideConfig()79 IndexUtil::GetDimensionStride(source_shape, minor_dimension); in StrideConfig()