Searched refs:tuple_param_shape (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder_test.cc | 611 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
613 auto p0 = Parameter(&b, 0, tuple_param_shape, "p0"); in TEST_F()
673 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
675 auto p0 = Parameter(&b, 0, tuple_param_shape, "p0"); in TEST_F()
691 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
694 auto p0 = Parameter(&b, 0, tuple_param_shape, "p0"); in TEST_F()
716 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
719 auto p0 = Parameter(&b, 0, tuple_param_shape, "p0"); in TEST_F()
742 Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
745 auto p0 = Parameter(&b, 0, tuple_param_shape, "p0"); in TEST_F()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_dataflow_analysis_test.cc | 1864 const Shape tuple_param_shape = ShapeUtil::MakeTupleShape( in TEST_P() local
1869 HloInstruction::CreateParameter(0, tuple_param_shape, "param_cond")); in TEST_P()
|