| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_inliner_test.cc | 148 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
150 empty.AddInstruction(HloInstruction::CreateParameter(0, r0s32, "A")); in TEST_F()
157 empty2.AddInstruction(HloInstruction::CreateParameter(0, r0s32, "A")); in TEST_F()
170 HloInstruction::CreateCall(r0s32, {zero}, empty_computation)); in TEST_F()
172 HloInstruction::CreateCall(r0s32, {zero}, empty2_computation)); in TEST_F()
174 HloInstruction::CreateCall(r0s32, {call1}, empty_computation)); in TEST_F()
|
| D | buffer_assignment_test.cc | 2206 const Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2273 const Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2356 const Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2364 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32, "x")); in TEST_F()
2377 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32, "x")); in TEST_F()
2379 HloInstruction::CreateBinary(r0s32, HloOpcode::kAdd, param, const9)); in TEST_F()
2389 builder.AddInstruction(HloInstruction::CreateInfeed(r0s32, token, "")); in TEST_F()
2391 HloInstruction::CreateGetTupleElement(r0s32, infeed, 0)); in TEST_F()
2395 HloInstruction::CreateWhile(r0s32, cond0, body0, infeed_data)); in TEST_F()
2400 HloInstruction::CreateWhile(r0s32, cond1, body1, while0)); in TEST_F()
[all …]
|
| D | hlo_computation_test.cc | 486 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
503 HloInstruction::CreateParameter(2, r0s32, "p.1")); in TEST_F()
514 ShapeUtil::Equal(clone->parameter_instruction(2)->shape(), r0s32)); in TEST_F()
|
| D | shape_inference_test.cc | 2306 auto r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2308 r0s32, in TEST_F()
2317 r0s32, in TEST_F()
2327 r0s32, {ShapeUtil::MakeProgramShape({tuple_f32_v32}, vector_32_)}, in TEST_F()
2343 r0s32, in TEST_F()
2354 r0s32, in TEST_F()
2355 {ShapeUtil::MakeProgramShape({r0s32}, f32_), in TEST_F()
2358 {r0s32, vector_32_, vector_64_}); in TEST_F()
2365 r0s32, in TEST_F()
2377 ShapeInference::InferConditionalShape(r0s32, {}, {}); in TEST_F()
[all …]
|
| D | algebraic_simplifier_test.cc | 366 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
369 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
373 HloInstruction::CreateBinary(r0s32, HloOpcode::kMultiply, param0, zero)); in TEST_F()
522 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
525 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
527 HloInstruction::CreateParameter(1, r0s32, "param1")); in TEST_F()
531 r0s32, HloOpcode::kSelect, one, param0, param1)); in TEST_F()
544 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
547 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
549 HloInstruction::CreateParameter(1, r0s32, "param1")); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | instruction_fusion_test.cc | 606 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
609 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
614 HloInstruction::CreateBinary(r0s32, HloOpcode::kDivide, param0, one)); in TEST_F()
616 HloInstruction::CreateBinary(r0s32, HloOpcode::kRemainder, param0, one)); in TEST_F()
|