Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc29 Shape MakeShapeWithLayout(PrimitiveType element_type, in MakeShapeWithLayout() function in xla::__anon17ef40ec0111::LayoutUtilTest
40 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 3}, {0, 1})}); in TEST_F()
42 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 2}, {0, 1})}); in TEST_F()
65 Shape src = MakeShapeWithLayout(F32, {2, 3}, {0, 1}); in TEST_F()
66 Shape dst = MakeShapeWithLayout(F32, {2, 3}, {1, 0}); in TEST_F()
89 {MakeShapeWithLayout(F32, {2, 3}, {0, 1}), in TEST_F()
90 MakeShapeWithLayout(F32, {42, 123}, {1, 0}), in TEST_F()
92 {MakeShapeWithLayout(F32, {}, {}), in TEST_F()
93 MakeShapeWithLayout(F32, {1, 2, 3}, {0, 2, 1})})}); in TEST_F()
95 {MakeShapeWithLayout(F32, {2, 3}, {1, 0}), in TEST_F()
[all …]
Dshape_test.cc41 ShapeUtil::MakeShapeWithLayout(F32, {}, {}, {Tile({256})});
43 const Shape matrix2_ = ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1});
96 EXPECT_NE(ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0}), in TEST_F()
97 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {0, 1})); in TEST_F()
100 EXPECT_NE(ShapeUtil::MakeShapeWithLayout(F32, {44, 23}, {1, 0}), in TEST_F()
101 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0})); in TEST_F()
104 EXPECT_NE(ShapeUtil::MakeShapeWithLayout(S32, {44, 23}, {1, 0}), in TEST_F()
105 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0})); in TEST_F()
108 EXPECT_EQ(ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0}), in TEST_F()
109 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0})); in TEST_F()
Dshape_util_test.cc114 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1})}), in TEST()
117 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1})}))); in TEST()
121 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1})}), in TEST()
124 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {1, 0})}))); in TEST()
164 ShapeUtil::MakeShapeWithLayout(F32, {4, 3}, {0, 1}), in TEST()
165 ShapeUtil::MakeShapeWithLayout(F16, {4, 3}, {0, 1}))); in TEST()
170 ShapeUtil::MakeShapeWithLayout(F32, {4, 3}, {0, 1}), in TEST()
171 ShapeUtil::MakeShapeWithLayout(F16, {3, 4}, {0, 1}))); in TEST()
173 ShapeUtil::MakeShapeWithLayout(F32, {3, 4}, {0, 1}), in TEST()
174 ShapeUtil::MakeShapeWithLayout(F16, {3, 4}, {1, 0}))); in TEST()
[all …]
Dliteral_util.h316 Literal literal(ShapeUtil::MakeShapeWithLayout( in CreateR2WithLayout()
405 Literal literal(ShapeUtil::MakeShapeWithLayout( in CreateFromArrayWithLayout()
Dliteral_test.cc386 Literal colmajor(ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1})); in TEST_F()
392 Literal rowmajor(ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0})); in TEST_F()
936 Shape shape = ShapeUtil::MakeShapeWithLayout( in TEST_F()
1142 Shape shape = ShapeUtil::MakeShapeWithLayout( in TEST_F()
1184 Shape shape = ShapeUtil::MakeShapeWithLayout( in TEST_F()
Dshape_util.cc216 /* static */ Shape ShapeUtil::MakeShapeWithLayout( in MakeShapeWithLayout() function in xla::ShapeUtil
229 return MakeShapeWithLayout(element_type, dimensions, layout); in MakeShapeWithDescendingLayout()
1426 Shape output_shape_with_layout = MakeShapeWithLayout( in AlignLayouts()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc66 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12}, {0}); in TEST_F()
68 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {24}, {0}); in TEST_F()
101 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12}, {0}); in TEST_F()
103 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {24}, {0}); 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 …]
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()
/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.cc173 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, {0, 1})), in XLA_TEST_F()
186 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, {1, 0})), in XLA_TEST_F()
265 {ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, in XLA_TEST_F()
267 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, in XLA_TEST_F()
593 ShapeUtil::MakeShapeWithLayout(F32, in XLA_TEST_F()
762 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{3}, {0}); in XLA_TEST_F()
799 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()
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/
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()
583 ShapeUtil::MakeShapeWithLayout(F32, {3, 5, 6, 7}, {2, 0, 3, 1}); in TEST_F()
721 {ShapeUtil::MakeShapeWithLayout(F32, {2, 2, 2}, {0, 1, 2}), in TEST_F()
723 ShapeUtil::MakeShapeWithLayout(F32, {2, 2, 2}, {1, 2, 0}), in TEST_F()
724 ShapeUtil::MakeShapeWithLayout(F32, {2, 2, 2}, {2, 0, 1}), in TEST_F()
784 Shape xshape = ShapeUtil::MakeShapeWithLayout(F32, {128, 8}, {0, 1}); in TEST_F()
862 {ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1})}); in TEST_F()
903 {ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1})}); in TEST_F()
942 Shape shape_copy = ShapeUtil::MakeShapeWithLayout(F32, {4, 5}, {1, 0}); in TEST_F()
[all …]
Dhlo_matchers_test.cc124 0, ShapeUtil::MakeShapeWithLayout(F32, {5, 7}, {0, 1}), "param"); in TEST_F()
140 op::Shape(ShapeUtil::MakeShapeWithLayout(F32, {5, 7}, {0, 1}))); in TEST_F()
142 EXPECT_THAT(p0.get(), op::ShapeWithLayout(ShapeUtil::MakeShapeWithLayout( in TEST_F()
147 ShapeUtil::MakeShapeWithLayout(F32, {5, 7}, {1, 0})))); in TEST_F()
154 Explain(p0.get(), op::ShapeWithLayout(ShapeUtil::MakeShapeWithLayout( in TEST_F()
271 Shape shape_memspace1 = ShapeUtil::MakeShapeWithLayout( in TEST_F()
274 Shape shape_memspace2 = ShapeUtil::MakeShapeWithLayout( in TEST_F()
Dpattern_matcher_test.cc570 auto shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 2}, {0, 1}); in TEST_F()
576 ShapeUtil::MakeShapeWithLayout(F32, {1, 2}, {1, 0}), in TEST_F()
627 ShapeUtil::MakeShapeWithLayout(F32, {1, 2}, {1, 0}), in TEST_F()
Dmemory_space_assignment_test.cc259 Shape shape_in_alternate_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P()
320 Shape shape_in_alternate_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P()
636 Shape shape_in_alternate_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P()
972 Shape shape_in_alternate_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P()
1027 Shape shape_in_alternate_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P()
2472 Shape shape_in_alternate_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P()
2542 Shape shape_in_default_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P()
2650 Shape shape_in_default_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P()
2654 Shape s32_in_default_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P()
2658 Shape f32v1_in_default_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P()
[all …]
Dalgebraic_simplifier_test.cc1776 0, ShapeUtil::MakeShapeWithLayout(F16, {1, 14, 14, 64}, {3, 2, 1, 0}), in TEST_F()
1802 0, ShapeUtil::MakeShapeWithLayout(F32, {1, 14, 14, 64}, {3, 2, 1, 0}), in TEST_F()
1829 0, ShapeUtil::MakeShapeWithLayout(F16, {1, 14, 14, 64}, {3, 2, 1, 0}), in TEST_F()
1884 0, ShapeUtil::MakeShapeWithLayout(F32, {1, 14, 14, 64}, {3, 2, 1, 0}), in TEST_F()
1887 ShapeUtil::MakeShapeWithLayout(F32, {1, 14, 14, 64}, {0, 1, 2, 3}), in TEST_F()
1891 ShapeUtil::MakeShapeWithLayout(F32, {14 * 14, 64}, {0, 1}), copy)); in TEST_F()
1893 ShapeUtil::MakeShapeWithLayout(F32, {14 * 14, 64}, {1, 0}), in TEST_F()
1913 0, ShapeUtil::MakeShapeWithLayout(F32, {1, 14, 14, 64}, {3, 2, 1, 0}), in TEST_F()
1916 ShapeUtil::MakeShapeWithLayout(F32, {1, 14, 14, 64}, {0, 1, 2, 3}), in TEST_F()
1919 ShapeUtil::MakeShapeWithLayout(F32, {14 * 14, 64}, {1, 0}), copy)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_ftz_test.cc37 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateBinaryOpModule()
54 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateUnaryOpModule()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc364 ShapeUtil::MakeShapeWithLayout(F32, {8, 8, 256, 64}, {3, 2, 1, 0}); in TEST_F()
401 Shape expected_shape = ShapeUtil::MakeShapeWithLayout(F32, {3, 2}, {1, 0}); in TEST_F()
428 Shape expected_shape = ShapeUtil::MakeShapeWithLayout(C64, {8, 32}, {1, 0}); in TEST_F()
Dreduction_layout_normalizer.cc99 Shape new_reduce_shape = ShapeUtil::MakeShapeWithLayout( in HandleReduce()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc121 return xla::ShapeUtil::MakeShapeWithLayout(type, dimensions, layout); in TensorShapeToXLAShape()
/external/tensorflow/tensorflow/compiler/xla/python/
Ddlpack.cc323 ShapeUtil::MakeShapeWithLayout(element_type, dimensions, minor_to_major); in DLPackManagedTensorToBuffer()

12