Searched refs:AreDimensionsConsecutive (Results 1 – 3 of 3) sorted by relevance
80 CHECK(LayoutUtil::AreDimensionsConsecutive(shape.layout(), dims_middle)); in PartitionShapeByMiddleDimensions()230 if (LayoutUtil::AreDimensionsConsecutive(input_shape.layout(), in GetReductionKindAndContiguousComponentsImpl()277 if (!LayoutUtil::AreDimensionsConsecutive(input->shape().layout(), in IsReductionFromOrToContiguousDimensions()279 !LayoutUtil::AreDimensionsConsecutive(input->shape().layout(), in IsReductionFromOrToContiguousDimensions()349 if (!LayoutUtil::AreDimensionsConsecutive(operand_shape.layout(), in IsReductionFromOrToContiguousDimensions()351 !LayoutUtil::AreDimensionsConsecutive(operand_shape.layout(), in IsReductionFromOrToContiguousDimensions()
181 static bool AreDimensionsConsecutive(const Layout& layout,
425 /* static */ bool LayoutUtil::AreDimensionsConsecutive( in AreDimensionsConsecutive() function in xla::LayoutUtil