Searched refs:tuple_param_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 486 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
488 auto p0 = Parameter(&b, 0, tuple_param_shape, "p0"); in TEST_F()
505 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
507 auto p0 = Parameter(&b, 0, tuple_param_shape, "p0"); in TEST_F()
523 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
526 auto p0 = Parameter(&b, 0, tuple_param_shape, "p0"); in TEST_F()
548 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
551 auto p0 = Parameter(&b, 0, tuple_param_shape, "p0"); in TEST_F()
574 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
577 auto p0 = Parameter(&b, 0, tuple_param_shape, "p0"); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis_test.cc | 1809 const Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_P() local
1814 HloInstruction::CreateParameter(0, tuple_param_shape, "param_cond")); in TEST_P()
|