Searched refs:starts_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis_test.cc | 627 Shape starts_shape = ShapeUtil::MakeShape(S32, {}); in Run() local
629 ShapeUtil::MakeTupleShape({input_shape, update_shape, starts_shape}); in Run()
658 HloInstruction::CreateGetTupleElement(starts_shape, tuple_param0, 2)); in Run()
1007 Shape starts_shape = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
1013 HloInstruction::CreateParameter(2, starts_shape, "starts")); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 2381 Shape starts_shape = ShapeUtil::MakeShape(S32, {1}); in TEST_F() local
2387 HloInstruction::CreateParameter(2, starts_shape, "start0")); in TEST_F()
2389 HloInstruction::CreateParameter(3, starts_shape, "start1")); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 335 Shape starts_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
342 HloInstruction::CreateParameter(1, starts_shape, "starts")); in TEST_F()
344 HloInstruction::CreateParameter(2, starts_shape, "starts")); in TEST_F()
|