Searched refs:GetWithDefaultLayout (Results 1 – 8 of 8) sorted by relevance
81 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape)); in TEST_F()83 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape)); in TEST_F()122 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape)); in TEST_F()124 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape)); in TEST_F()166 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_a_shape)); in TEST_F()168 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_b_shape)); in TEST_F()170 ShapeLayout(LayoutUtil::GetWithDefaultLayout(tuple_result->shape())); in TEST_F()200 ShapeLayout(LayoutUtil::GetWithDefaultLayout(rhs_shape)); in TEST_F()202 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape)); in TEST_F()240 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape)); in TEST_F()[all …]
51 CHECK(shape == LayoutUtil::GetWithDefaultLayout(shape)) in HandleReduce()
48 CHECK(shape == LayoutUtil::GetWithDefaultLayout(shape)) in HandleReduce()
70 static Shape GetWithDefaultLayout(const Shape& shape);
162 /* static */ Shape LayoutUtil::GetWithDefaultLayout(const Shape& shape) { in GetWithDefaultLayout() function in xla::LayoutUtil
248 xla::LayoutUtil::GetWithDefaultLayout( in CopyDeviceTensorToCPU()
422 return LayoutUtil::GetWithDefaultLayout(host_shape); in ChooseCompactLayoutForShape()
606 xla::LayoutUtil::GetWithDefaultLayout(*subshape), output_tensor, in Compute()