Home
last modified time | relevance | path

Searched refs:IsMonotonicWithDim0Major (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util.h94 static bool IsMonotonicWithDim0Major(const Layout& layout);
Dlayout_util.cc259 /* static */ bool LayoutUtil::IsMonotonicWithDim0Major(const Layout& layout) { in IsMonotonicWithDim0Major() function in xla::LayoutUtil
Dliteral_util.cc742 if (!LayoutUtil::IsMonotonicWithDim0Major(shape().layout())) { in Reshape()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc862 if (LayoutUtil::IsMonotonicWithDim0Major(lhs_shape.layout()) && in HandleConvolution()
863 LayoutUtil::IsMonotonicWithDim0Major(rhs_shape.layout()) && in HandleConvolution()
864 LayoutUtil::IsMonotonicWithDim0Major(convolution_shape.layout())) { in HandleConvolution()
1149 TF_RET_CHECK(LayoutUtil::IsMonotonicWithDim0Major(operand->shape().layout())); in HandleFft()
1150 TF_RET_CHECK(LayoutUtil::IsMonotonicWithDim0Major(fft->shape().layout())); in HandleFft()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc678 LayoutUtil::IsMonotonicWithDim0Major(operand->shape().layout())); in CheckCustomCallLayout()
682 LayoutUtil::IsMonotonicWithDim0Major(custom_call->shape().layout())); in CheckCustomCallLayout()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc486 LayoutUtil::IsMonotonicWithDim0Major(fft->operand(0)->shape().layout())); in HandleFft()
487 TF_RET_CHECK(LayoutUtil::IsMonotonicWithDim0Major(fft->shape().layout())); in HandleFft()