Home
last modified time | relevance | path

Searched refs:MakeTupleShape (Results 1 – 25 of 119) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/xla/service/
Doptimize_input_output_buffer_alias_test.cc74 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_}); in TEST_F()
75 Shape output = ShapeUtil::MakeTupleShape({r3f32_, r4f32_}); in TEST_F()
83 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
84 Shape output = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
98 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r1f32_, r2f32_, r2f32_}); in TEST_F()
99 Shape output = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
112 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
113 Shape output = ShapeUtil::MakeTupleShape({r4f32_, r3f32_, r2f32_, r1f32_}); in TEST_F()
126 Shape input = ShapeUtil::MakeTupleShape( in TEST_F()
127 {ShapeUtil::MakeTupleShape({r1f32_}), r2f32_, r3f32_, r4f32_}); in TEST_F()
[all …]
Dhlo_sharding_test.cc155 HloSharding sharding = HloSharding::SingleTuple(ShapeUtil::MakeTupleShape({}), in TEST_F()
162 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape({ in TEST_F()
164 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {3})}), in TEST_F()
185 EXPECT_IS_NOT_OK(tuple_sharding.Validate(ShapeUtil::MakeTupleShape({}), in TEST_F()
226 ShapeTree<HloSharding> shape_tree(ShapeUtil::MakeTupleShape({}), in TEST_F()
234 ShapeTree<HloSharding> shape_tree(ShapeUtil::MakeTupleShape({}), in TEST_F()
243 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
247 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
257 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
261 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
[all …]
Dshaped_buffer_test.cc119 s = xla::ShapeUtil::MakeTupleShape(std::vector<xla::Shape>(2, s)); in TEST()
120 s = xla::ShapeUtil::MakeTupleShape(std::vector<xla::Shape>(3, s)); in TEST()
157 xla::ShapeUtil::MakeTupleShape({array_shape, array_shape}); in TEST()
184 shape = xla::ShapeUtil::MakeTupleShape(shapes); in BM_TakeSubTree()
Dwhile_loop_invariant_code_motion_test.cc64 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, scalar_s32}); in TEST_F()
106 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, scalar_s32}); in TEST_F()
178 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); in TEST_F()
217 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, scalar_s32}); in TEST_F()
256 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, token_shape}); in TEST_F()
305 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, token_shape}); in TEST_F()
358 Shape while_shape = ShapeUtil::MakeTupleShape( in TEST_F()
409 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, scalar_s32}); in TEST_F()
445 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); in TEST_F()
Dcopy_insertion_test.cc307 ShapeUtil::MakeTupleShape( in TEST_F()
308 {ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {}), in TEST_F()
343 ShapeUtil::MakeTupleShape( in TEST_F()
344 {ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {}), in TEST_F()
476 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in BuildDependentBodyComputation2()
782 ShapeUtil::MakeTupleShape({induction_variable_shape_, data_shape_});
784 ShapeUtil::MakeTupleShape({data_shape_, data_shape_});
785 Shape nested_loop_state_shape_ = ShapeUtil::MakeTupleShape(
1448 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in TEST_F()
1510 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
[all …]
Dbatchnorm_expander_test.cc72 ShapeUtil::MakeTupleShape({input_shape, scale_shape, offset_shape}), in TEST_F()
115 ShapeUtil::MakeTupleShape({input_shape, scale_shape, mean_shape}), param0, in TEST_F()
Dshape_inference_test.cc114 Shape tuple = ShapeUtil::MakeTupleShape({s32_, f32_}); in TEST_F()
162 HloOpcode::kSelect, ShapeUtil::MakeTupleShape({pred_, pred_}), in TEST_F()
163 ShapeUtil::MakeTupleShape({f32_, f32_}), in TEST_F()
164 ShapeUtil::MakeTupleShape({f32_, f32_})); in TEST_F()
311 ShapeUtil::MakeTupleShape({s32_, f32_}))); in TEST_F()
1094 {f32_, s32_, f32_, s32_}, ShapeUtil::MakeTupleShape({f32_, s32_})); in TEST_F()
1098 EXPECT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeTupleShape({f32_, s32_}), in TEST_F()
1108 {f32_, s32_, f32_, s32_}, ShapeUtil::MakeTupleShape({f32_, s32_})); in TEST_F()
1123 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {5, 2, 0}), in TEST_F()
1133 ShapeUtil::MakeTupleShape({f32_, s32_})); in TEST_F()
[all …]
Dbfloat16_normalization_test.cc260 ShapeUtil::MakeTupleShape({f32_shape, bf16_shape}), {a, b}, reduction, in TEST_F()
291 ShapeUtil::MakeTupleShape({bf16_shape, bf16_shape}), {a, a}, in TEST_F()
320 ShapeUtil::MakeTupleShape({bf16_shape, f32_shape}), {a, a}, in TEST_F()
349 MakeSortHlo(ShapeUtil::MakeTupleShape({bf16_shape, s32_shape}), in TEST_F()
377 MakeSortHlo(ShapeUtil::MakeTupleShape({bf16_shape, f32_shape}), in TEST_F()
Dlayout_assignment_test.cc275 ShapeUtil::MakeTupleShape({constant0->shape(), constant1->shape()}); in TEST_F()
505 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
720 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
722 ShapeUtil::MakeTupleShape({ in TEST_F()
755 Shape tshape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_F()
756 Shape result_tshape = ShapeUtil::MakeTupleShape({shape}); in TEST_F()
861 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
902 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
1286 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
1335 ShapeLayout(ShapeUtil::MakeTupleShape({ashape_major, ashape_minor})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc102 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}), in TEST()
103 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}))); in TEST()
114 ShapeUtil::MakeTupleShape( in TEST()
117 ShapeUtil::MakeTupleShape( in TEST()
121 ShapeUtil::MakeTupleShape( in TEST()
124 ShapeUtil::MakeTupleShape( in TEST()
226 Shape tuple1 = ShapeUtil::MakeTupleShape( in TEST()
228 Shape tuple2 = ShapeUtil::MakeTupleShape( in TEST()
234 Shape tuple1 = ShapeUtil::MakeTupleShape( in TEST()
236 Shape tuple2 = ShapeUtil::MakeTupleShape( in TEST()
[all …]
Dlayout_util_test.cc40 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 3}, {0, 1})}); in TEST_F()
42 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 2}, {0, 1})}); in TEST_F()
44 Shape tuple0 = ShapeUtil::MakeTupleShape({}); in TEST_F()
45 Shape tuple1 = ShapeUtil::MakeTupleShape({shape}); in TEST_F()
46 Shape tuple2 = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_F()
58 Shape other_tuple2 = ShapeUtil::MakeTupleShape({shape, other_shape}); in TEST_F()
88 Shape src = ShapeUtil::MakeTupleShape( in TEST_F()
91 ShapeUtil::MakeTupleShape( in TEST_F()
94 Shape dst = ShapeUtil::MakeTupleShape( in TEST_F()
97 ShapeUtil::MakeTupleShape( in TEST_F()
[all …]
Dshape_tree_test.cc32 ShapeUtil::MakeTupleShape({array_shape_, array_shape_, array_shape_}); in ShapeTreeTest()
33 nested_tuple_shape_ = ShapeUtil::MakeTupleShape( in ShapeTreeTest()
34 {array_shape_, ShapeUtil::MakeTupleShape({array_shape_, array_shape_}), in ShapeTreeTest()
35 ShapeUtil::MakeTupleShape( in ShapeTreeTest()
36 {ShapeUtil::MakeTupleShape({array_shape_, array_shape_}), in ShapeTreeTest()
121 ShapeTree<int> shape_tree{ShapeUtil::MakeTupleShape({})}; in TEST_F()
305 ShapeUtil::MakeTupleShape({array_shape_, array_shape_})); in TEST_F()
335 ShapeUtil::MakeTupleShape({array_shape_, array_shape_}), 0); in TEST_F()
545 shape = ShapeUtil::MakeTupleShape(shapes); in BM_Construct()
560 shape = ShapeUtil::MakeTupleShape(shapes); in BM_ConstructUnowned()
[all …]
Dshape_test.cc46 ShapeUtil::MakeTupleShape({opaque_, scalar_, matrix_, matrix2_});
48 ShapeUtil::MakeTupleShape({tuple_, matrix_, token_});
150 *program_shape.add_parameters() = ShapeUtil::MakeTupleShape( in TEST_F()
152 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_bytesizeof_test.cc26 xla::ShapeUtil::MakeTupleShape({xla::ShapeUtil::MakeShape(xla::F32, {})}); in TEST_F()
34 xla::ShapeUtil::MakeTupleShape({xla::ShapeUtil::MakeShape(xla::F32, {})}); in TEST_F()
Dcpu_infeed_test.cc200 const auto infeed1_shape = ShapeUtil::MakeTupleShape( in TEST_F()
202 const auto infeed2_shape = ShapeUtil::MakeTupleShape( in TEST_F()
204 const auto result_shape = ShapeUtil::MakeTupleShape( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/python/
Doutfeed_receiver_test.cc137 EXPECT_EQ(ShapeUtil::MakeTupleShape({shape0}), received[0].data->shape()); in TEST()
180 EXPECT_EQ(ShapeUtil::MakeTupleShape({shape0}), received[0].data->shape()); in TEST()
182 EXPECT_EQ(ShapeUtil::MakeTupleShape({shape1}), received[1].data->shape()); in TEST()
223 EXPECT_EQ(ShapeUtil::MakeTupleShape({shape0}), received[0].data->shape()); in TEST()
225 EXPECT_EQ(ShapeUtil::MakeTupleShape({shape1}), received[1].data->shape()); in TEST()
318 EXPECT_EQ(ShapeUtil::MakeTupleShape({shape0}), received[0].data->shape()); in TEST()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc558 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
603 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
621 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
646 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
672 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
694 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
715 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
738 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
766 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
808 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dwhile_test.cc360 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()
414 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()
471 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()
520 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()
566 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()
625 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()
704 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()
769 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()
836 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()
1042 auto result_shape = ShapeUtil::MakeTupleShape( in XLA_TEST_F()
[all …]
Dlocal_client_execute_test.cc264 Shape shape_with_layout = ShapeUtil::MakeTupleShape( in XLA_TEST_F()
286 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
288 ShapeUtil::MakeTupleShape({vector_shape, array_shape}); in XLA_TEST_F()
332 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
334 ShapeUtil::MakeTupleShape({inner_tuple_shape, vector_shape}); in XLA_TEST_F()
373 ShapeUtil::MakeTupleShape({array_shape, array_shape}); in XLA_TEST_F()
416 const Shape tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F()
456 const Shape inner_tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F()
458 const Shape tuple_shape = ShapeUtil::MakeTupleShape(inner_tuple_shapes); in XLA_TEST_F()
513 shape = ShapeUtil::MakeTupleShape({shape}); in XLA_TEST_F()
Dconditional_test.cc165 Shape tuple_2_r0f32_ = ShapeUtil::MakeTupleShape(
167 Shape tuple_2_r1s2f32_ = ShapeUtil::MakeTupleShape(
169 Shape empty_tuple_ = ShapeUtil::MakeTupleShape({});
635 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F()
665 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F()
705 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0f32_, r0f32_}); in XLA_TEST_F()
760 const Shape tuple2 = ShapeUtil::MakeTupleShape({scalar, scalar}); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcustom_call_test.cc153 ShapeUtil::MakeTupleShape({ in TEST_F()
155 ShapeUtil::MakeTupleShape({ in TEST_F()
Dgpu_layout_assignment_test.cc194 Shape batchnorm_shape = ShapeUtil::MakeTupleShape( in TEST_F()
235 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
267 ShapeUtil::MakeTupleShape({shape, scale_shape, scale_shape}); in TEST_F()
317 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
Dwhile_transformer_test.cc111 return ShapeUtil::MakeTupleShape( in GetLoopStateShape()
114 return ShapeUtil::MakeTupleShape( in GetLoopStateShape()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dqr.cc77 Shape qr_shape = ShapeUtil::MakeTupleShape({q_shape, a_shape}); in QRDecomposition()
Dlu_decomposition.cc43 Shape lu_shape = ShapeUtil::MakeTupleShape( in LuDecomposition()

12345