| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_computation_test.cc | 54 HloInstruction::CreateParameter(0, r0f32_, "param0")); in CreateNegateComputation()
56 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, param)); in CreateNegateComputation()
65 HloInstruction::CreateParameter(0, r0f32_, "param0")); in CreateMapComputation()
67 HloInstruction::CreateMap(r0f32_, {param}, map_computation)); in CreateMapComputation()
71 Shape r0f32_ = ShapeUtil::MakeShape(F32, {}); member in xla::__anond1f842fb0111::HloComputationTest
105 HloInstruction::CreateParameter(0, r0f32_, "param0")); in TEST_F()
107 HloInstruction::CreateMap(r0f32_, {param}, map1_computation)); in TEST_F()
109 HloInstruction::CreateMap(r0f32_, {param}, map2_computation)); in TEST_F()
111 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, map1, map2)); in TEST_F()
141 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, constant)); in TEST_F()
[all …]
|
| D | hlo_instruction_test.cc | 45 Shape r0f32_ = ShapeUtil::MakeShape(F32, {}); member in xla::__anon9b186d4a0111::HloInstructionTest
136 auto parameter = HloInstruction::CreateParameter(1, r0f32_, "foo"); in TEST_F()
150 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
152 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "bar")); in TEST_F()
154 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, bar)); in TEST_F()
183 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
185 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "bar")); in TEST_F()
187 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
189 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
191 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, bar)); in TEST_F()
[all …]
|
| D | hlo_module_test.cc | 56 HloInstruction::CreateCall(r0f32_, {}, computation)); in CreateCallComputation()
61 Shape r0f32_ = ShapeUtil::MakeShape(F32, {}); member in xla::__anon943d828e0111::HloModuleTest
120 auto x = b.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in TEST_F()
122 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, x, x)); in TEST_F()
132 HloInstruction::CreateFusion(r0f32_, HloInstruction::FusionKind::kInput, in TEST_F()
|
| D | hlo_subcomputation_unification_test.cc | 62 Shape r0f32_ = ShapeUtil::MakeShape(S32, {}); member in xla::HloSubcomputationUnificationTest
160 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in TEST_F()
162 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "y")); in TEST_F()
164 HloInstruction::CreateBinary(r0f32_, HloOpcode::kPower, x, y)); in TEST_F()
|
| D | buffer_assignment_test.cc | 158 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildMapComputationPlus1()
162 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param, value)); in BuildMapComputationPlus1()
169 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildReduceComputation()
171 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "y")); in BuildReduceComputation()
173 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param, param2)); in BuildReduceComputation()
235 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildR0F32UnaryOpComputation()
236 builder.AddInstruction(HloInstruction::CreateUnary(r0f32_, opcode, param)); in BuildR0F32UnaryOpComputation()
300 Shape r0f32_ = ShapeUtil::MakeShape(xla::F32, {}); member in xla::__anon3198ac380111::BufferAssignmentTest
430 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "p")); in TEST_F()
513 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "p")); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | call_test.cc | 38 Parameter(&builder, 0, r0f32_, "x"); in CreateR0F32IdentityComputation()
66 Tuple(&builder, {Parameter(&builder, 0, r0f32_, "x")}); in CreateR0F32TupleComputation()
72 Shape r0f32_ = ShapeUtil::MakeShape(F32, {}); member in xla::__anonb8db6b040111::CallOpTest
111 auto x = Parameter(&builder, 0, r0f32_, "x"); in XLA_TEST_F()
118 auto x = Parameter(&builder2, 0, r0f32_, "x"); in XLA_TEST_F()
127 auto x = Parameter(&builder3, 0, r0f32_, "x"); in XLA_TEST_F()
|
| D | conditional_test.cc | 39 Parameter(&builder, 0, r0f32_, "x"); in CreateR0IdentityComputation()
55 return CreateCeilComputation(r0f32_); in CreateR0CeilComputation()
72 return CreateFloorComputation(r0f32_); in CreateR0FloorComputation()
163 Shape r0f32_ = ShapeUtil::MakeShape(F32, {}); member in xla::__anon97db189c0111::ConditionalOpTest
247 Parameter(sb.get(), 0, r0f32_, "p0")); in XLA_TEST_P()
346 auto true_operand = Parameter(&inner_builder, 1, r0f32_, "param1"); in XLA_TEST_F()
347 auto false_operand = Parameter(&inner_builder, 2, r0f32_, "param2"); in XLA_TEST_F()
635 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F()
665 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F()
705 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0f32_, r0f32_}); in XLA_TEST_F()
|
| D | custom_call_test.cc | 77 Shape r0f32_ = ShapeUtil::MakeShape(F32, {}); member in xla::__anon047ef51e0211::CustomCallTest
88 HloInstruction::CreateCustomCall(r0f32_, {constant}, "R0F32Add2")); in XLA_TEST_F()
109 HloInstruction::CreateCustomCall(r0f32_, {constant}, "R2F32ReduceSum")); in XLA_TEST_F()
|