Searched refs:IsMonotonicWithDim0Major (Results 1 – 10 of 10) sorted by relevance
181 TF_RET_CHECK(LayoutUtil::IsMonotonicWithDim0Major( in CudaArrayInterface()258 !LayoutUtil::IsMonotonicWithDim0Major(shape.layout())) { in PjRtBufferGetBuffer()264 !LayoutUtil::IsMonotonicWithDim0Major(shape.layout()) && in PjRtBufferGetBuffer()
91 if (xla::LayoutUtil::IsMonotonicWithDim0Major( in TransposeTensor()204 if (!xla::LayoutUtil::IsMonotonicWithDim0Major(shape.layout())) { in AutoTransposeAndLinearize()434 if (!xla::LayoutUtil::IsMonotonicWithDim0Major(xla_shape_.layout())) { in DoWork()495 if (!xla::LayoutUtil::IsMonotonicWithDim0Major( in DoWork()
110 static bool IsMonotonicWithDim0Major(const Layout& layout);
289 /* static */ bool LayoutUtil::IsMonotonicWithDim0Major(const Layout& layout) { in IsMonotonicWithDim0Major() function in xla::LayoutUtil
827 if (!LayoutUtil::IsMonotonicWithDim0Major(shape().layout())) { in Reshape()
885 if (LayoutUtil::IsMonotonicWithDim0Major(lhs_shape.layout()) && in HandleConvolution()886 LayoutUtil::IsMonotonicWithDim0Major(rhs_shape.layout()) && in HandleConvolution()887 LayoutUtil::IsMonotonicWithDim0Major(convolution_shape.layout())) { in HandleConvolution()1026 TF_RET_CHECK(LayoutUtil::IsMonotonicWithDim0Major(operand->shape().layout())); in HandleFft()1027 TF_RET_CHECK(LayoutUtil::IsMonotonicWithDim0Major(fft->shape().layout())); in HandleFft()2262 TF_RET_CHECK(LayoutUtil::IsMonotonicWithDim0Major( in HandleTopK()2264 TF_RET_CHECK(LayoutUtil::IsMonotonicWithDim0Major( in HandleTopK()2267 LayoutUtil::IsMonotonicWithDim0Major(hlo->operand(0)->shape().layout())); in HandleTopK()
1083 LayoutUtil::IsMonotonicWithDim0Major(shape.layout())); in CollapseFirstNDims()
199 if (!LayoutUtil::IsMonotonicWithDim0Major(instr->shape().layout())) { in HasOnlyRowMajorLayout()
1573 TF_RET_CHECK(LayoutUtil::IsMonotonicWithDim0Major(operand_shape.layout())); in EmitFftThunkFromMlir()1574 TF_RET_CHECK(LayoutUtil::IsMonotonicWithDim0Major(output_shape.layout())); in EmitFftThunkFromMlir()
50 LayoutUtil::IsMonotonicWithDim0Major(shape.layout())) { in GetPermutationIfAvailable()