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