Home
last modified time | relevance | path

Searched refs:GetDefaultLayoutForShape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util.h57 static Layout GetDefaultLayoutForShape(const Shape& shape);
Dlayout_util_test.cc237 LayoutUtil::GetDefaultLayoutForShape(shape.tuple_shapes(0)), in TEST_F()
250 LayoutUtil::GetDefaultLayoutForShape( in TEST_F()
Dlayout_util.cc117 /* static */ Layout LayoutUtil::GetDefaultLayoutForShape(const Shape& shape) { in GetDefaultLayoutForShape() function in xla::LayoutUtil
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.h343 return LayoutUtil::GetDefaultLayoutForShape(buffer.shape()); in GetUnconstrainedLayout()
Dlayout_assignment.cc1082 LayoutUtil::GetDefaultLayoutForShape(operand_shape); in ChooseOperandLayoutFromOutputLayout()
1143 LayoutUtil::GetDefaultLayoutForShape(output_shape); in ChooseOutputLayoutFromOperandLayout()