Home
last modified time | relevance | path

Searched refs:r0s32 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment_test.cc1980 const Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2047 const Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2130 const Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2138 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32, "x")); in TEST_F()
2151 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32, "x")); in TEST_F()
2153 HloInstruction::CreateBinary(r0s32, HloOpcode::kAdd, param, const9)); in TEST_F()
2163 builder.AddInstruction(HloInstruction::CreateInfeed(r0s32, token, "")); in TEST_F()
2165 HloInstruction::CreateGetTupleElement(r0s32, infeed, 0)); in TEST_F()
2169 HloInstruction::CreateWhile(r0s32, cond0, body0, infeed_data)); in TEST_F()
2174 HloInstruction::CreateWhile(r0s32, cond1, body1, while0)); in TEST_F()
[all …]
Dshape_inference_test.cc1625 auto r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
1627 r0s32, in TEST_F()
1636 r0s32, in TEST_F()
1646 r0s32, {ShapeUtil::MakeProgramShape({tuple_f32_v32}, vector_32_)}, in TEST_F()
1662 r0s32, in TEST_F()
1673 r0s32, in TEST_F()
1674 {ShapeUtil::MakeProgramShape({r0s32}, f32_), in TEST_F()
1677 {r0s32, vector_32_, vector_64_}); in TEST_F()
1684 r0s32, in TEST_F()
1696 ShapeInference::InferConditionalShape(r0s32, {}, {}); in TEST_F()
Dhlo_computation_test.cc504 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()
Dalgebraic_simplifier_test.cc281 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
284 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
288 HloInstruction::CreateBinary(r0s32, HloOpcode::kMultiply, param0, zero)); in TEST_F()
300 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
303 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
305 HloInstruction::CreateParameter(1, r0s32, "param1")); in TEST_F()
309 r0s32, HloOpcode::kSelect, one, param0, param1)); in TEST_F()
322 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
325 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
327 HloInstruction::CreateParameter(1, r0s32, "param1")); in TEST_F()
[all …]