Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc88 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {1572864}), in TEST()
99 Shape expected = ShapeUtil::MakeTupleShape({ in TEST()
101 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {2})}), in TEST()
192 Shape tuple1 = ShapeUtil::MakeTupleShape( in TEST()
194 Shape tuple2 = ShapeUtil::MakeTupleShape( in TEST()
200 Shape tuple1 = ShapeUtil::MakeTupleShape( in TEST()
202 Shape tuple2 = ShapeUtil::MakeTupleShape( in TEST()
208 Shape tuple1 = ShapeUtil::MakeTupleShape( in TEST()
210 Shape tuple2 = ShapeUtil::MakeTupleShape( in TEST()
217 Shape tuple1 = 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.cc30 ShapeUtil::MakeTupleShape({array_shape_, array_shape_, array_shape_}); in ShapeTreeTest()
31 nested_tuple_shape_ = ShapeUtil::MakeTupleShape( in ShapeTreeTest()
32 {array_shape_, ShapeUtil::MakeTupleShape({array_shape_, array_shape_}), in ShapeTreeTest()
33 ShapeUtil::MakeTupleShape( in ShapeTreeTest()
34 {ShapeUtil::MakeTupleShape({array_shape_, array_shape_}), in ShapeTreeTest()
296 ShapeUtil::MakeTupleShape({array_shape_, array_shape_})); in TEST_F()
326 ShapeUtil::MakeTupleShape({array_shape_, array_shape_}), 0); in TEST_F()
Dshape_util.h303 static Shape MakeTupleShape(tensorflow::gtl::ArraySlice<Shape> shapes);
316 static Shape MakeNil() { return MakeTupleShape({}); } in MakeNil()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_test.cc142 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape({ in TEST_F()
144 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {3})}), in TEST_F()
166 EXPECT_IS_NOT_OK(tuple_sharding.Validate(ShapeUtil::MakeTupleShape({}), in TEST_F()
228 ShapeTree<HloSharding> shape_tree(ShapeUtil::MakeTupleShape({}), in TEST_F()
236 ShapeTree<HloSharding> shape_tree(ShapeUtil::MakeTupleShape({}), in TEST_F()
245 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
249 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
259 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
263 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {4})}), in TEST_F()
Dwhile_loop_invariant_code_motion_test.cc62 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, scalar_s32}); in TEST_F()
104 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, scalar_s32}); in TEST_F()
176 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); in TEST_F()
214 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, scalar_s32}); in TEST_F()
250 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); in TEST_F()
289 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); in TEST_F()
332 ShapeUtil::MakeTupleShape({scalar_s32, scalar_f32, scalar_f32}); in TEST_F()
381 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, scalar_s32}); in TEST_F()
416 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); in TEST_F()
Dbatchnorm_expander_test.cc58 ShapeUtil::MakeTupleShape({input_shape, scale_shape, offset_shape}), in TEST_F()
100 ShapeUtil::MakeTupleShape({input_shape, scale_shape, mean_shape}), param0, in TEST_F()
Dcopy_insertion_test.cc301 ShapeUtil::MakeTupleShape( in TEST_F()
302 {ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {}), in TEST_F()
337 ShapeUtil::MakeTupleShape( in TEST_F()
338 {ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {}), in TEST_F()
467 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in BuildDependentBodyComputation2()
765 ShapeUtil::MakeTupleShape({induction_variable_shape_, data_shape_});
767 ShapeUtil::MakeTupleShape({data_shape_, data_shape_});
768 Shape nested_loop_state_shape_ = ShapeUtil::MakeTupleShape(
1227 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in TEST_F()
1290 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
[all …]
Dshape_inference_test.cc111 Shape tuple = ShapeUtil::MakeTupleShape({s32_, f32_}); in TEST_F()
155 TernaryOperation::TRIOP_SELECT, ShapeUtil::MakeTupleShape({pred_, pred_}), in TEST_F()
156 ShapeUtil::MakeTupleShape({f32_, f32_}), in TEST_F()
157 ShapeUtil::MakeTupleShape({f32_, f32_})); in TEST_F()
299 ShapeUtil::MakeTupleShape({s32_, f32_}))); in TEST_F()
794 Shape tuple_shape = ShapeUtil::MakeTupleShape({f32_, s32_}); in TEST_F()
1224 Shape result_shape = ShapeUtil::MakeTupleShape({s32_, vector_32_}); in TEST_F()
1236 Shape result_shape = ShapeUtil::MakeTupleShape({s32_, vector_32_}); in TEST_F()
1312 Shape tuple = ShapeUtil::MakeTupleShape({vector_32_}); in TEST_F()
1390 Shape tuple_shape = ShapeUtil::MakeTupleShape({input_shape, input_shape}); in TEST_F()
[all …]
Dlayout_assignment_test.cc246 ShapeUtil::MakeTupleShape({constant0->shape(), constant1->shape()}); in TEST_F()
478 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
619 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F()
621 ShapeUtil::MakeTupleShape({ in TEST_F()
665 Shape tshape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_F()
666 Shape result_tshape = ShapeUtil::MakeTupleShape({shape}); in TEST_F()
Dhlo_scheduling_test.cc39 ShapeUtil::MakeTupleShape({scalar_shape, scalar_shape}); in TEST_F()
Dbuffer_assignment_test.cc246 Shape t_s32_f32v4_ = ShapeUtil::MakeTupleShape({s32_, f32vec4_});
247 Shape t_s32_f32v10_ = ShapeUtil::MakeTupleShape({s32_, f32vec10_});
1108 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(PRED, {1, 2, 3, 4}), in TEST_F()
1140 ShapeUtil::MakeTupleShape( in TEST_F()
1142 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(S32, {42}), in TEST_F()
1202 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(PRED, {1, 2, 3, 4}), in TEST_F()
1222 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F()
1261 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F()
1345 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(PRED, {1, 2, 3, 4})}); in TEST_F()
1525 ShapeUtil::MakeTupleShape({data_shape_, data_shape_, data_shape_});
Dliveness_util_test.cc59 0, ShapeUtil::MakeTupleShape({elem_shape, elem_shape}), "tuple")); in TEST_F()
87 0, ShapeUtil::MakeTupleShape({data_shape, data_shape}), "tuple")); in TEST_F()
201 0, ShapeUtil::MakeTupleShape({data_shape, data_shape}), "tuple")); in TEST_F()
Dhlo_alias_analysis_test.cc328 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
422 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
494 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_}); in TEST_F()
592 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_, scalar_shape_}); in TEST_F()
740 const Shape tuple_shape = ShapeUtil::MakeTupleShape({scalar_shape_}); in TEST_F()
Dbuffer_liveness_test.cc480 ShapeUtil::MakeTupleShape( in TEST_F()
544 ShapeUtil::MakeTupleShape( in TEST_F()
620 0, ShapeUtil::MakeTupleShape({data_shape, data_shape}), "param0")); in Run()
750 0, ShapeUtil::MakeTupleShape({data_shape, data_shape}), "param0")); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfusion_merger_test.cc113 Shape tuple_shape2_ = ShapeUtil::MakeTupleShape({data_shape_, data_shape_}); in BuildComputation1()
232 Shape tuple_shape2_ = ShapeUtil::MakeTupleShape({data_shape_, data_shape_});
234 ShapeUtil::MakeTupleShape({data_shape_, data_shape_, data_shape_});
235 Shape tuple_shape4_ = ShapeUtil::MakeTupleShape(
Dgpu_layout_assignment_test.cc183 Shape batchnorm_shape = ShapeUtil::MakeTupleShape( in TEST_F()
224 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
254 ShapeUtil::MakeTupleShape({shape, scale_shape, scale_shape}); in TEST_F()
304 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
Dwhile_transformer_test.cc128 return ShapeUtil::MakeTupleShape( in GetLoopStateShape()
131 return ShapeUtil::MakeTupleShape( in GetLoopStateShape()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc273 Shape shape_with_layout = ShapeUtil::MakeTupleShape( in XLA_TEST_F()
295 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
297 ShapeUtil::MakeTupleShape({vector_shape, array_shape}); in XLA_TEST_F()
342 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
344 ShapeUtil::MakeTupleShape({inner_tuple_shape, vector_shape}); in XLA_TEST_F()
385 ShapeUtil::MakeTupleShape({array_shape, array_shape}); in XLA_TEST_F()
431 const Shape tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F()
478 const Shape inner_tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F()
480 const Shape tuple_shape = ShapeUtil::MakeTupleShape(inner_tuple_shapes); in XLA_TEST_F()
535 shape = ShapeUtil::MakeTupleShape({shape}); in XLA_TEST_F()
Dwhile_test.cc364 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in TEST_F()
417 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in TEST_F()
473 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in TEST_F()
522 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in TEST_F()
569 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in TEST_F()
630 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in TEST_F()
709 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in TEST_F()
776 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in TEST_F()
843 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()
1059 auto result_shape = ShapeUtil::MakeTupleShape( in TEST_F()
[all …]
Dconditional_test.cc163 Shape tuple_2_r0f32_ = ShapeUtil::MakeTupleShape(
165 Shape tuple_2_r1s2f32_ = ShapeUtil::MakeTupleShape(
167 Shape empty_tuple_ = ShapeUtil::MakeTupleShape({});
505 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F()
535 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); 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()
467 Shape arg0_shape = ShapeUtil::MakeTupleShape( in XLA_TEST_F()
468 {c64r0, ShapeUtil::MakeTupleShape({c64r1, c64r2})}); in XLA_TEST_F()
Dclient_test.cc83 ShapeUtil::MakeTupleShape( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_copy_insertion_test.cc111 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_, scalar_shape_}); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dwhile_loop.cc35 xla::Shape tuple_shape = xla::ShapeUtil::MakeTupleShape(var_shapes); in XlaWhileLoop()

123