Home
last modified time | relevance | path

Searched refs:MakeShapeWithLayout (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc29 Shape MakeShapeWithLayout(PrimitiveType element_type, in MakeShapeWithLayout() function in xla::__anon976bbfea0111::LayoutUtilTest
48 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 3}, {0, 1})}); in TEST_F()
50 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 2}, {0, 1})}); in TEST_F()
73 Shape src = MakeShapeWithLayout(F32, {2, 3}, {0, 1}); in TEST_F()
74 Shape dst = MakeShapeWithLayout(F32, {2, 3}, {1, 0}); in TEST_F()
97 Shape dst = MakeShapeWithLayout(F32, {2, 3}, {1, 0}); in TEST_F()
120 {MakeShapeWithLayout(F32, {2, 3}, {0, 1}), in TEST_F()
121 MakeShapeWithLayout(F32, {42, 123}, {1, 0}), in TEST_F()
123 {MakeShapeWithLayout(F32, {}, {}), in TEST_F()
124 MakeShapeWithLayout(F32, {1, 2, 3}, {0, 2, 1})})}); in TEST_F()
[all …]
Dshape_test.cc39 ShapeUtil::MakeShapeWithLayout(F32, {}, {}, {Tile({256})});
41 const Shape matrix2_ = ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1});
94 EXPECT_NE(ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0}), in TEST_F()
95 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {0, 1})); in TEST_F()
98 EXPECT_NE(ShapeUtil::MakeShapeWithLayout(F32, {44, 23}, {1, 0}), in TEST_F()
99 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0})); in TEST_F()
102 EXPECT_NE(ShapeUtil::MakeShapeWithLayout(S32, {44, 23}, {1, 0}), in TEST_F()
103 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0})); in TEST_F()
106 EXPECT_EQ(ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0}), in TEST_F()
107 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0})); in TEST_F()
Dshape_util_test.cc113 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1})}), in TEST()
116 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1})}))); in TEST()
120 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1})}), in TEST()
123 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {1, 0})}))); in TEST()
163 ShapeUtil::MakeShapeWithLayout(F32, {4, 3}, {0, 1}), in TEST()
164 ShapeUtil::MakeShapeWithLayout(F16, {4, 3}, {0, 1}))); in TEST()
169 ShapeUtil::MakeShapeWithLayout(F32, {4, 3}, {0, 1}), in TEST()
170 ShapeUtil::MakeShapeWithLayout(F16, {3, 4}, {0, 1}))); in TEST()
172 ShapeUtil::MakeShapeWithLayout(F32, {3, 4}, {0, 1}), in TEST()
173 ShapeUtil::MakeShapeWithLayout(F16, {3, 4}, {1, 0}))); in TEST()
[all …]
Dliteral_util.h334 Literal literal(ShapeUtil::MakeShapeWithLayout( in CreateR2WithLayout()
438 Literal literal(ShapeUtil::MakeShapeWithLayout( in CreateFromArrayWithLayout()
Dshape_util.cc189 /* static */ Shape ShapeUtil::MakeShapeWithLayout( in MakeShapeWithLayout() function in xla::ShapeUtil
202 return MakeShapeWithLayout(element_type, dimensions, layout); in MakeShapeWithDescendingLayout()
1399 Shape output_shape_with_layout = MakeShapeWithLayout( in AlignLayouts()
Dliteral_test.cc423 Literal colmajor(ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1})); in TEST_F()
429 Literal rowmajor(ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0})); in TEST_F()
973 Shape shape = ShapeUtil::MakeShapeWithLayout( in TEST_F()
1179 Shape shape = ShapeUtil::MakeShapeWithLayout( in TEST_F()
1221 Shape shape = ShapeUtil::MakeShapeWithLayout( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dshape_partition_test.cc47 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {1, 3}, {1, 0}), 2, in TEST_F()
55 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {3, 1}, {0, 1}), 2, in TEST_F()
63 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {5, 3}, {1, 0}), 6, in TEST_F()
70 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {5, 3}, {0, 1}), 4, in TEST_F()
80 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {5, 3, 2}, {2, 1, 0}), 16, in TEST_F()
90 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {5, 3, 2}, {2, 0, 1}), 16, in TEST_F()
100 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {5, 3}, {1, 0}); in TEST_F()
125 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {5, 3, 2}, {2, 1, 0}); in TEST_F()
165 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, RandR4Dims(), {3, 2, 1, 0}); in TEST_F()
Dcpu_layout_assignment_test.cc66 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1}); in TEST_F()
68 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1}); in TEST_F()
101 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1}); in TEST_F()
103 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1}); in TEST_F()
143 Shape lhs_a_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1}); in TEST_F()
144 Shape lhs_b_shape = ShapeUtil::MakeShapeWithLayout(F32, {2, 12}, {0, 1}); in TEST_F()
145 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1}); in TEST_F()
146 Shape result_a_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1}); in TEST_F()
147 Shape result_b_shape = ShapeUtil::MakeShapeWithLayout(F32, {2, 24}, {0, 1}); in TEST_F()
185 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1}); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc311 ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0}); in TEST_F()
313 ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1}); in TEST_F()
586 ShapeUtil::MakeShapeWithLayout(F32, {3, 5, 6, 7}, {2, 0, 3, 1}); in TEST_F()
724 {ShapeUtil::MakeShapeWithLayout(F32, {2, 2, 2}, {0, 1, 2}), in TEST_F()
726 ShapeUtil::MakeShapeWithLayout(F32, {2, 2, 2}, {1, 2, 0}), in TEST_F()
727 ShapeUtil::MakeShapeWithLayout(F32, {2, 2, 2}, {2, 0, 1}), in TEST_F()
787 Shape xshape = ShapeUtil::MakeShapeWithLayout(F32, {128, 8}, {0, 1}); in TEST_F()
865 {ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1})}); in TEST_F()
879 ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0}))); in TEST_F()
909 {ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1})}); in TEST_F()
[all …]
Dhlo_matchers_test.cc109 0, ShapeUtil::MakeShapeWithLayout(F32, {5, 7}, {0, 1}), "param"); in TEST()
125 op::Shape(ShapeUtil::MakeShapeWithLayout(F32, {5, 7}, {0, 1}))); in TEST()
127 EXPECT_THAT(p0.get(), op::ShapeWithLayout(ShapeUtil::MakeShapeWithLayout( in TEST()
132 ShapeUtil::MakeShapeWithLayout(F32, {5, 7}, {1, 0})))); in TEST()
139 Explain(p0.get(), op::ShapeWithLayout(ShapeUtil::MakeShapeWithLayout( in TEST()
Dpattern_matcher_test.cc569 auto shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 2}, {0, 1}); in TEST()
575 ShapeUtil::MakeShapeWithLayout(F32, {1, 2}, {1, 0}), in TEST()
626 ShapeUtil::MakeShapeWithLayout(F32, {1, 2}, {1, 0}), in TEST()
Dalgebraic_simplifier_test.cc1569 0, ShapeUtil::MakeShapeWithLayout(F32, {1, 14, 14, 64}, {3, 2, 1, 0}), in TEST_F()
1572 ShapeUtil::MakeShapeWithLayout(F32, {1, 14, 14, 64}, {0, 1, 2, 3}), in TEST_F()
1576 ShapeUtil::MakeShapeWithLayout(F32, {14 * 14, 64}, {0, 1}), copy)); in TEST_F()
1578 ShapeUtil::MakeShapeWithLayout(F32, {14 * 14, 64}, {1, 0}), in TEST_F()
1598 0, ShapeUtil::MakeShapeWithLayout(F32, {1, 14, 14, 64}, {3, 2, 1, 0}), in TEST_F()
1601 ShapeUtil::MakeShapeWithLayout(F32, {1, 14, 14, 64}, {0, 1, 2, 3}), in TEST_F()
1604 ShapeUtil::MakeShapeWithLayout(F32, {14 * 14, 64}, {1, 0}), copy)); in TEST_F()
1624 0, ShapeUtil::MakeShapeWithLayout(F32, {1, 14, 14, 64}, {0, 1, 2, 3}), in TEST_F()
1627 ShapeUtil::MakeShapeWithLayout(F32, {1, 14, 14, 64}, {1, 2, 0, 3}), in TEST_F()
2195 ShapeUtil::MakeShapeWithLayout(F32, {2, 2, 2}, {0, 1, 2}), in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_test.cc52 ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2}, in XLA_TEST_F()
85 {ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2}, in XLA_TEST_F()
87 ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2}, in XLA_TEST_F()
104 ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2}, in XLA_TEST_F()
108 ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2}, in XLA_TEST_F()
Dlocal_client_execute_test.cc172 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, {0, 1})), in XLA_TEST_F()
184 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, {1, 0})), in XLA_TEST_F()
262 {ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, in XLA_TEST_F()
264 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, in XLA_TEST_F()
590 ShapeUtil::MakeShapeWithLayout(F32, in XLA_TEST_F()
759 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{3}, {0}); in XLA_TEST_F()
Dcopy_test.cc245 Shape in_shape = ShapeUtil::MakeShapeWithLayout(F32, {0, 0}, {0, 1}); in XLA_TEST_F()
246 Shape out_shape = ShapeUtil::MakeShapeWithLayout(F32, {0, 0}, {1, 0}); in XLA_TEST_F()
Dmultioutput_fusion_test.cc67 const Shape elem_shape0 = ShapeUtil::MakeShapeWithLayout(F32, {}, {}); in RunTest2D()
69 ShapeUtil::MakeShapeWithLayout(F32, {size, size}, {1, 0}); in RunTest2D()
Dcustom_call_test.cc174 ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0}); in XLA_TEST_F()
Dbroadcast_simple_test.cc61 *r3_shape = ShapeUtil::MakeShapeWithLayout(F32, bounds, minor_to_major); in MakeR3Data()
75 *r2_shape = ShapeUtil::MakeShapeWithLayout(F32, bounds, minor_to_major); in MakeR2Data()
Ddot_operation_test.cc321 ShapeUtil::MakeShapeWithLayout( in TestImpl()
326 ShapeUtil::MakeShapeWithLayout( in TestImpl()
335 ShapeUtil::MakeShapeWithLayout( in TestImpl()
Dtransfer_manager_test.cc158 ShapeUtil::MakeShapeWithLayout(F32, {2, 3}, {1, 0}); in XLA_TEST_F()
Dreshape_test.cc620 ShapeUtil::MakeShapeWithLayout(use_bfloat16() ? BF16 : F32, {2, 8}, in XLA_TEST_P()
770 ShapeUtil::MakeShapeWithLayout(use_bfloat16() ? BF16 : F32, {7, 2, 3, 5}, in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_ftz_test.cc34 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateBinaryOpModule()
51 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateUnaryOpModule()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc362 ShapeUtil::MakeShapeWithLayout(F32, {8, 8, 256, 64}, {3, 2, 1, 0}); in TEST_F()
399 Shape expected_shape = ShapeUtil::MakeShapeWithLayout(F32, {3, 2}, {1, 0}); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc121 return xla::ShapeUtil::MakeShapeWithLayout(type, dimensions, layout); in TensorShapeToXLAShape()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc174 xla::ShapeUtil::MakeShapeWithLayout(xla::F32, {2, 2}, {0, 1}), "P0"); in Dot()
177 xla::ShapeUtil::MakeShapeWithLayout(xla::F32, {2, 1}, {0, 1}), "P1"); in Dot()
873 xla::ShapeUtil::MakeShapeWithLayout(xla::F32, {2, 2}, {0, 1}).ToProto(); in TEST()
875 xla::ShapeUtil::MakeShapeWithLayout(xla::F32, {2, 1}, {0, 1}).ToProto(); in TEST()
877 xla::ShapeUtil::MakeShapeWithLayout(xla::F32, {2, 1}, {0, 1}).ToProto(); in TEST()

12