Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc99 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}), in TEST()
100 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}))); in TEST()
111 ShapeUtil::MakeTupleShape( in TEST()
114 ShapeUtil::MakeTupleShape( in TEST()
118 ShapeUtil::MakeTupleShape( in TEST()
121 ShapeUtil::MakeTupleShape( in TEST()
202 Shape tuple1 = ShapeUtil::MakeTupleShape( in TEST()
204 Shape tuple2 = ShapeUtil::MakeTupleShape( in TEST()
210 Shape tuple1 = ShapeUtil::MakeTupleShape( in TEST()
212 Shape tuple2 = ShapeUtil::MakeTupleShape( in TEST()
[all …]
Dlayout_util_test.cc48 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 3}, {0, 1})}); in TEST_F()
50 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 2}, {0, 1})}); in TEST_F()
52 Shape tuple0 = ShapeUtil::MakeTupleShape({}); in TEST_F()
53 Shape tuple1 = ShapeUtil::MakeTupleShape({shape}); in TEST_F()
54 Shape tuple2 = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_F()
66 Shape other_tuple2 = ShapeUtil::MakeTupleShape({shape, other_shape}); in TEST_F()
119 Shape src = ShapeUtil::MakeTupleShape( in TEST_F()
122 ShapeUtil::MakeTupleShape( in TEST_F()
125 Shape dst = ShapeUtil::MakeTupleShape( in TEST_F()
128 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()
519 shape = ShapeUtil::MakeTupleShape(shapes); in BM_Construct()
533 shape = ShapeUtil::MakeTupleShape(shapes); in BM_ConstructUnowned()
[all …]
Dshape_test.cc43 ShapeUtil::MakeTupleShape({opaque_, scalar_, matrix_, matrix2_});
45 ShapeUtil::MakeTupleShape({tuple_, matrix_, token_});
147 *program_shape.add_parameters() = ShapeUtil::MakeTupleShape( in TEST_F()
149 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}), in TEST_F()
Dliteral_util.cc422 Literal literal(ShapeUtil::MakeTupleShape(element_shapes)); in MakeTuple()
435 Literal literal(ShapeUtil::MakeTupleShape(element_shapes)); in MakeTupleFromSlices()
449 Literal literal(ShapeUtil::MakeTupleShape(element_shapes)); in MakeTupleOwned()
/external/tensorflow/tensorflow/compiler/xla/service/
Doptimize_input_output_buffer_alias_test.cc79 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_}); in TEST_F()
80 Shape output = ShapeUtil::MakeTupleShape({r3f32_, r4f32_}); in TEST_F()
88 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
89 Shape output = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
103 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r1f32_, r2f32_, r2f32_}); in TEST_F()
104 Shape output = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
117 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
118 Shape output = ShapeUtil::MakeTupleShape({r4f32_, r3f32_, r2f32_, r1f32_}); in TEST_F()
131 Shape input = ShapeUtil::MakeTupleShape( in TEST_F()
132 {ShapeUtil::MakeTupleShape({r1f32_}), r2f32_, r3f32_, r4f32_}); in TEST_F()
[all …]
Dhlo_sharding_test.cc120 HloSharding sharding = HloSharding::SingleTuple(ShapeUtil::MakeTupleShape({}), in TEST_F()
127 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape({ in TEST_F()
129 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {3})}), in TEST_F()
150 EXPECT_IS_NOT_OK(tuple_sharding.Validate(ShapeUtil::MakeTupleShape({}), in TEST_F()
191 ShapeTree<HloSharding> shape_tree(ShapeUtil::MakeTupleShape({}), in TEST_F()
199 ShapeTree<HloSharding> shape_tree(ShapeUtil::MakeTupleShape({}), in TEST_F()
208 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
212 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
222 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
226 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
[all …]
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()
Dshaped_buffer_test.cc116 s = xla::ShapeUtil::MakeTupleShape(std::vector<xla::Shape>(2, s)); in TEST()
117 s = xla::ShapeUtil::MakeTupleShape(std::vector<xla::Shape>(3, s)); in TEST()
159 shape = xla::ShapeUtil::MakeTupleShape(shapes); in BM_TakeSubTree()
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()
473 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in BuildDependentBodyComputation2()
776 ShapeUtil::MakeTupleShape({induction_variable_shape_, data_shape_});
778 ShapeUtil::MakeTupleShape({data_shape_, data_shape_});
779 Shape nested_loop_state_shape_ = ShapeUtil::MakeTupleShape(
1238 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in TEST_F()
1300 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
[all …]
Dshape_inference_test.cc112 Shape tuple = ShapeUtil::MakeTupleShape({s32_, f32_}); in TEST_F()
156 HloOpcode::kSelect, ShapeUtil::MakeTupleShape({pred_, pred_}), in TEST_F()
157 ShapeUtil::MakeTupleShape({f32_, f32_}), in TEST_F()
158 ShapeUtil::MakeTupleShape({f32_, f32_})); in TEST_F()
299 ShapeUtil::MakeTupleShape({s32_, f32_}))); in TEST_F()
717 {f32_, s32_, f32_, s32_}, ShapeUtil::MakeTupleShape({f32_, s32_})); in TEST_F()
721 EXPECT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeTupleShape({f32_, s32_}), in TEST_F()
730 ShapeUtil::MakeTupleShape({f32_, s32_})); in TEST_F()
742 {s32_, s32_, f32_, s32_}, ShapeUtil::MakeTupleShape({f32_, s32_})); in TEST_F()
754 {s32_, s32_, f32_, s32_}, ShapeUtil::MakeTupleShape({f32_, s32_})); 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()
Dlayout_assignment_test.cc275 ShapeUtil::MakeTupleShape({constant0->shape(), constant1->shape()}); in TEST_F()
505 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
723 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
725 ShapeUtil::MakeTupleShape({ in TEST_F()
758 Shape tshape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_F()
759 Shape result_tshape = ShapeUtil::MakeTupleShape({shape}); in TEST_F()
864 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
908 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
1292 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
1341 ShapeLayout(ShapeUtil::MakeTupleShape({ashape_major, ashape_minor})); in TEST_F()
Dhlo_alias_analysis_test.cc223 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
275 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
336 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
524 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
619 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
692 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
791 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_, scalar_shape_}); in TEST_F()
940 const Shape tuple_shape = ShapeUtil::MakeTupleShape({scalar_shape_}); in TEST_F()
Dbfloat16_normalization_test.cc258 ShapeUtil::MakeTupleShape({f32_shape, bf16_shape}), {a, b}, reduction, in TEST_F()
288 MakeSortHlo(ShapeUtil::MakeTupleShape({bf16_shape, s32_shape}), in TEST_F()
317 MakeSortHlo(ShapeUtil::MakeTupleShape({bf16_shape, f32_shape}), in TEST_F()
Dpattern_matcher_test.cc130 auto tuple_shape = ShapeUtil::MakeTupleShape({ in TEST()
643 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {2, 2})}), in TEST()
658 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {2, 2})}), in TEST()
667 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTupleShape({shape})}), in TEST()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc486 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
505 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
523 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
548 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
574 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
596 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
617 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
640 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
668 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
710 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
[all …]
/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/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.cc261 Shape shape_with_layout = ShapeUtil::MakeTupleShape( in XLA_TEST_F()
283 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
285 ShapeUtil::MakeTupleShape({vector_shape, array_shape}); in XLA_TEST_F()
329 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
331 ShapeUtil::MakeTupleShape({inner_tuple_shape, vector_shape}); in XLA_TEST_F()
370 ShapeUtil::MakeTupleShape({array_shape, array_shape}); in XLA_TEST_F()
413 const Shape tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F()
453 const Shape inner_tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F()
455 const Shape tuple_shape = ShapeUtil::MakeTupleShape(inner_tuple_shapes); in XLA_TEST_F()
510 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()
Dtuple_test.cc434 Shape inner_tuple_shape = ShapeUtil::MakeTupleShape({data_shape, data_shape}); in XLA_TEST_F()
436 ShapeUtil::MakeTupleShape({inner_tuple_shape, data_shape}); in XLA_TEST_F()
465 Shape arg0_shape = ShapeUtil::MakeTupleShape( in XLA_TEST_F()
466 {c64r0, ShapeUtil::MakeTupleShape({c64r1, c64r2})}); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc192 Shape batchnorm_shape = ShapeUtil::MakeTupleShape( in TEST_F()
233 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
265 ShapeUtil::MakeTupleShape({shape, scale_shape, scale_shape}); in TEST_F()
315 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
Dwhile_transformer_test.cc111 return ShapeUtil::MakeTupleShape( in GetLoopStateShape()
114 return ShapeUtil::MakeTupleShape( in GetLoopStateShape()

1234