Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc100 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}), in TEST()
101 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}))); in TEST()
112 ShapeUtil::MakeTupleShape( in TEST()
115 ShapeUtil::MakeTupleShape( in TEST()
119 ShapeUtil::MakeTupleShape( in TEST()
122 ShapeUtil::MakeTupleShape( in TEST()
224 Shape tuple1 = ShapeUtil::MakeTupleShape( in TEST()
226 Shape tuple2 = ShapeUtil::MakeTupleShape( in TEST()
232 Shape tuple1 = ShapeUtil::MakeTupleShape( in TEST()
234 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()
519 shape = ShapeUtil::MakeTupleShape(shapes); in BM_Construct()
533 shape = ShapeUtil::MakeTupleShape(shapes); in BM_ConstructUnowned()
[all …]
Dshape_test.cc45 ShapeUtil::MakeTupleShape({opaque_, scalar_, matrix_, matrix2_});
47 ShapeUtil::MakeTupleShape({tuple_, matrix_, token_});
149 *program_shape.add_parameters() = ShapeUtil::MakeTupleShape( in TEST_F()
151 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Doptimize_input_output_buffer_alias_test.cc81 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_}); in TEST_F()
82 Shape output = ShapeUtil::MakeTupleShape({r3f32_, r4f32_}); in TEST_F()
90 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
91 Shape output = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
105 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r1f32_, r2f32_, r2f32_}); in TEST_F()
106 Shape output = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
119 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
120 Shape output = ShapeUtil::MakeTupleShape({r4f32_, r3f32_, r2f32_, r1f32_}); in TEST_F()
133 Shape input = ShapeUtil::MakeTupleShape( in TEST_F()
134 {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 …]
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()
183 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(
1249 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in TEST_F()
1311 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.cc113 Shape tuple = ShapeUtil::MakeTupleShape({s32_, f32_}); in TEST_F()
161 HloOpcode::kSelect, ShapeUtil::MakeTupleShape({pred_, pred_}), in TEST_F()
162 ShapeUtil::MakeTupleShape({f32_, f32_}), in TEST_F()
163 ShapeUtil::MakeTupleShape({f32_, f32_})); in TEST_F()
310 ShapeUtil::MakeTupleShape({s32_, f32_}))); in TEST_F()
907 {f32_, s32_, f32_, s32_}, ShapeUtil::MakeTupleShape({f32_, s32_})); in TEST_F()
911 EXPECT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeTupleShape({f32_, s32_}), in TEST_F()
920 ShapeUtil::MakeTupleShape({f32_, s32_})); in TEST_F()
932 {s32_, s32_, f32_, s32_}, ShapeUtil::MakeTupleShape({f32_, s32_})); in TEST_F()
944 {s32_, s32_, f32_, s32_}, ShapeUtil::MakeTupleShape({f32_, s32_})); in TEST_F()
[all …]
Dbfloat16_normalization_test.cc260 ShapeUtil::MakeTupleShape({f32_shape, bf16_shape}), {a, b}, reduction, in TEST_F()
290 ShapeUtil::MakeTupleShape({bf16_shape, bf16_shape}), {a, a}, in TEST_F()
319 ShapeUtil::MakeTupleShape({bf16_shape, f32_shape}), {a, a}, in TEST_F()
348 MakeSortHlo(ShapeUtil::MakeTupleShape({bf16_shape, s32_shape}), in TEST_F()
376 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()
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()
/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/client/
Dxla_builder_test.cc496 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
515 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
533 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
558 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
584 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
606 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
627 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
650 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
678 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
720 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcustom_call_test.cc131 ShapeUtil::MakeTupleShape({ in TEST_F()
133 ShapeUtil::MakeTupleShape({ in TEST_F()
160 auto tuple_shape = 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/python/
Dshared_device_buffer_test.cc47 Shape tuple_shape = ShapeUtil::MakeTupleShape({a_shape, b_shape}); in TEST()
75 Shape ab_tuple_shape = ShapeUtil::MakeTupleShape({a_shape, b_shape}); in TEST()
77 Shape abc_tuple_shape = ShapeUtil::MakeTupleShape({c_shape, ab_tuple_shape}); in TEST()
/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/mlir/tensorflow/utils/
Dcompile_mlir_util_test.cc92 xla::ShapeUtil::MakeTupleShape({xla::ShapeUtil::MakeShape(xla::F32, {}), in TEST()
100 xla::ShapeUtil::MakeTupleShape({output_shape}); in TEST()

1234