| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_computation_test.cc | 53 HloInstruction::CreateParameter(0, r0f32_, "param0")); in CreateNegateComputation()
55 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, param)); in CreateNegateComputation()
64 HloInstruction::CreateParameter(0, r0f32_, "param0")); in CreateMapComputation()
66 HloInstruction::CreateMap(r0f32_, {param}, map_computation)); in CreateMapComputation()
70 Shape r0f32_ = ShapeUtil::MakeShape(F32, {}); member in xla::__anone2bdf1820111::HloComputationTest
104 HloInstruction::CreateParameter(0, r0f32_, "param0")); in TEST_F()
106 HloInstruction::CreateMap(r0f32_, {param}, map1_computation)); in TEST_F()
108 HloInstruction::CreateMap(r0f32_, {param}, map2_computation)); in TEST_F()
110 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, map1, map2)); in TEST_F()
140 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, constant)); in TEST_F()
[all …]
|
| D | hlo_instruction_test.cc | 47 Shape r0f32_ = ShapeUtil::MakeShape(F32, {}); member in xla::__anonabde1bd10111::HloInstructionTest
138 auto parameter = HloInstruction::CreateParameter(1, r0f32_, "foo"); in TEST_F()
152 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
154 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "bar")); in TEST_F()
156 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, bar)); in TEST_F()
185 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
187 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "bar")); in TEST_F()
189 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
191 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
193 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, bar)); in TEST_F()
[all …]
|
| D | hlo_module_test.cc | 57 HloInstruction::CreateCall(r0f32_, {}, computation)); in CreateCallComputation()
62 Shape r0f32_ = ShapeUtil::MakeShape(F32, {}); member in xla::__anon3ac340b50111::HloModuleTest
121 auto x = b.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in TEST_F()
123 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, x, x)); in TEST_F()
133 HloInstruction::CreateFusion(r0f32_, HloInstruction::FusionKind::kInput, in TEST_F()
373 b.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "p0")); in TEST_F()
375 b.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "p1")); in TEST_F()
376 b.AddInstruction(HloInstruction::CreateParameter(2, r0f32_, "p2")); in TEST_F()
377 b.AddInstruction(HloInstruction::CreateParameter(3, r0f32_, "p3")); 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 | 165 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildMapComputationPlus1()
169 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param, value)); in BuildMapComputationPlus1()
176 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildReduceComputation()
178 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "y")); in BuildReduceComputation()
180 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param, param2)); in BuildReduceComputation()
242 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildR0F32UnaryOpComputation()
243 builder.AddInstruction(HloInstruction::CreateUnary(r0f32_, opcode, param)); in BuildR0F32UnaryOpComputation()
307 Shape r0f32_ = ShapeUtil::MakeShape(xla::F32, {}); member in xla::__anon8a84187f0111::BufferAssignmentTest
437 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "p")); in TEST_F()
519 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::__anon17d4f22b0111::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::__anonf9fe2de30111::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::__anon66a20a650211::CustomCallTest
88 HloInstruction::CreateCustomCall(r0f32_, {constant}, "R0F32Add2")); in XLA_TEST_F()
109 HloInstruction::CreateCustomCall(r0f32_, {constant}, "R2F32ReduceSum")); in XLA_TEST_F()
|
| D | buffer_donation_test.cc | 245 Shape r0f32_ = ShapeUtil::MakeShape(xla::F32, {}); member in xla::__anon352f73640111::BufferDonationTest
|