Searched refs:r0s32_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_subcomputation_unification_test.cc | 34 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32_, "x")); in CreateR0S32IdentityComputation()
41 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32_, "x")); in CreateR0S32AdditionComputation()
43 builder.AddInstruction(HloInstruction::CreateParameter(1, r0s32_, "y")); in CreateR0S32AdditionComputation()
45 HloInstruction::CreateBinary(r0s32_, HloOpcode::kAdd, x, y)); in CreateR0S32AdditionComputation()
61 Shape r0s32_ = ShapeUtil::MakeShape(S32, {}); member in xla::HloSubcomputationUnificationTest
79 HloInstruction::CreateCall(r0s32_, {constant}, callee1)); in TEST_F()
81 HloInstruction::CreateCall(r0s32_, {constant}, callee2)); in TEST_F()
83 HloInstruction::CreateBinary(r0s32_, HloOpcode::kAdd, x, y)); in TEST_F()
108 HloInstruction::CreateCall(r0s32_, {constant1, constant2}, callee1)); in TEST_F()
110 HloInstruction::CreateCall(r0s32_, {constant1, constant2}, callee2)); in TEST_F()
[all …]
|