| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_inliner_test.cc | 149 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
151 empty.AddInstruction(HloInstruction::CreateParameter(0, r0s32, "A")); in TEST_F()
158 empty2.AddInstruction(HloInstruction::CreateParameter(0, r0s32, "A")); in TEST_F()
171 HloInstruction::CreateCall(r0s32, {zero}, empty_computation)); in TEST_F()
173 HloInstruction::CreateCall(r0s32, {zero}, empty2_computation)); in TEST_F()
175 HloInstruction::CreateCall(r0s32, {call1}, empty_computation)); in TEST_F()
|
| D | buffer_assignment_test.cc | 2128 const Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2195 const Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2278 const Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2286 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32, "x")); in TEST_F()
2299 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32, "x")); in TEST_F()
2301 HloInstruction::CreateBinary(r0s32, HloOpcode::kAdd, param, const9)); in TEST_F()
2311 builder.AddInstruction(HloInstruction::CreateInfeed(r0s32, token, "")); in TEST_F()
2313 HloInstruction::CreateGetTupleElement(r0s32, infeed, 0)); in TEST_F()
2317 HloInstruction::CreateWhile(r0s32, cond0, body0, infeed_data)); in TEST_F()
2322 HloInstruction::CreateWhile(r0s32, cond1, body1, while0)); in TEST_F()
[all …]
|
| D | hlo_computation_test.cc | 504 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
521 HloInstruction::CreateParameter(2, r0s32, "p.1")); in TEST_F()
532 ShapeUtil::Equal(clone->parameter_instruction(2)->shape(), r0s32)); in TEST_F()
|
| D | shape_inference_test.cc | 2274 auto r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2276 r0s32, in TEST_F()
2285 r0s32, in TEST_F()
2295 r0s32, {ShapeUtil::MakeProgramShape({tuple_f32_v32}, vector_32_)}, in TEST_F()
2311 r0s32, in TEST_F()
2322 r0s32, in TEST_F()
2323 {ShapeUtil::MakeProgramShape({r0s32}, f32_), in TEST_F()
2326 {r0s32, vector_32_, vector_64_}); in TEST_F()
2333 r0s32, in TEST_F()
2345 ShapeInference::InferConditionalShape(r0s32, {}, {}); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 365 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
368 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
372 HloInstruction::CreateBinary(r0s32, HloOpcode::kMultiply, param0, zero)); in TEST_F()
521 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
524 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
526 HloInstruction::CreateParameter(1, r0s32, "param1")); in TEST_F()
530 r0s32, HloOpcode::kSelect, one, param0, param1)); in TEST_F()
543 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
546 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
548 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()
|