Home
last modified time | relevance | path

Searched refs:CreateUnary (Results 1 – 25 of 51) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc65 HloInstruction::CreateUnary(vshape, HloOpcode::kNegate, add1)); in TEST_F()
99 HloInstruction::CreateUnary(vshape, HloOpcode::kNegate, input)); in TEST_F()
101 HloInstruction::CreateUnary(vshape, HloOpcode::kCeil, negate)); in TEST_F()
103 HloInstruction::CreateUnary(vshape, HloOpcode::kExp, ceil)); in TEST_F()
105 HloInstruction::CreateUnary(vshape, HloOpcode::kFloor, exp)); in TEST_F()
149 HloInstruction::CreateUnary(vshape, HloOpcode::kNegate, input)); in TEST_F()
151 HloInstruction::CreateUnary(vshape, HloOpcode::kCeil, negate)); in TEST_F()
179 HloInstruction::CreateUnary(cshape, HloOpcode::kExp, reduce)); in TEST_F()
181 HloInstruction::CreateUnary(cshape, HloOpcode::kFloor, exp)); in TEST_F()
246 HloInstruction::CreateUnary(vshape, HloOpcode::kNegate, constant)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dreduce_precision_insertion_test.cc55 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
82 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
141 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
170 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
172 HloInstruction::CreateUnary(shape, HloOpcode::kSin, a)); in TEST_F()
206 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
235 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
240 HloInstruction::CreateUnary(shape, HloOpcode::kCos, b)); in TEST_F()
270 HloInstruction::CreateUnary(shape, HloOpcode::kCos, x)); in TEST_F()
296 HloInstruction::CreateUnary(shape, HloOpcode::kCos, x)); in TEST_F()
[all …]
Dbuffer_liveness_test.cc114 HloInstruction::CreateUnary(vec_, HloOpcode::kNegate, param)); in TEST_F()
116 HloInstruction::CreateUnary(vec_, HloOpcode::kExp, negate)); in TEST_F()
118 HloInstruction::CreateUnary(vec_, HloOpcode::kLog, exp)); in TEST_F()
161 HloInstruction::CreateUnary(vec_, HloOpcode::kNegate, param0)); in TEST_F()
163 HloInstruction::CreateUnary(vec_, HloOpcode::kExp, param1)); in TEST_F()
210 HloInstruction::CreateUnary(vec_, HloOpcode::kExp, param)); in TEST_F()
212 HloInstruction::CreateUnary(vec_, HloOpcode::kNegate, exp)); in TEST_F()
244 HloInstruction::CreateUnary(vec_, HloOpcode::kNegate, param)); in TEST_F()
246 HloInstruction::CreateUnary(vec_, HloOpcode::kExp, param)); in TEST_F()
286 HloInstruction::CreateUnary(vec_, HloOpcode::kNegate, param)); in TEST_F()
[all …]
Dhlo_memory_scheduler_test.cc57 HloInstruction::CreateUnary(vec, HloOpcode::kAbs, param)); in TEST_F()
59 HloInstruction::CreateUnary(vec, HloOpcode::kExp, param)); in TEST_F()
64 HloInstruction::CreateUnary(vec, HloOpcode::kNegate, exp)); in TEST_F()
160 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, lit)); in TEST_F()
163 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, abs_const)); in TEST_F()
170 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, abs_const)); in TEST_F()
217 HloInstruction::CreateUnary(r1f32, HloOpcode::kExp, c3)); in TEST_F()
Dhlo_dce_test.cc104 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
108 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
133 auto dead_negate = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
141 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
185 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, param)); in TEST_F()
197 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, param)); in TEST_F()
252 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, param)); in TEST_F()
307 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, param)); in TEST_F()
Dhlo_rematerialization_test.cc68 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kNegate, bcast)); in MakeRematerializableComputation()
357 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kTanh, rng)); in TEST_F()
359 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kExp, rng)); in TEST_F()
511 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kCopy, param)); in TEST_F()
514 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kNegate, copy)); in TEST_F()
516 auto negate_a_2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
520 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kNegate, copy)); in TEST_F()
522 auto negate_b_2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
614 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kNegate, gte)); in TEST_P()
Dhlo_cse_test.cc227 auto exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
229 auto exp2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
231 auto exp3 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
431 auto exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
435 auto exp2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
462 auto exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
466 auto exp2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
540 auto negate1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
542 auto exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
547 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
[all …]
Dhlo_instruction_test.cc187 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
189 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
300 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, c0)); in TEST_F()
308 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, addtotal)); in TEST_F()
480 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
482 HloInstruction::CreateUnary(r0f32_, HloOpcode::kLog, foo)); in TEST_F()
548 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
610 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
612 HloInstruction::CreateUnary(r0f32_, HloOpcode::kLog, foo)); in TEST_F()
632 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, constant)); in TEST_F()
[all …]
Ddefuser_test.cc95 HloInstruction::CreateUnary(shape_, HloOpcode::kNegate, add)); in TEST_F()
123 HloInstruction::CreateUnary(shape_, HloOpcode::kNegate, add)); in TEST_F()
162 HloInstruction::CreateUnary(shape_, HloOpcode::kNegate, add)); in TEST_F()
200 HloInstruction::CreateUnary(shape_, HloOpcode::kNegate, add)); in TEST_F()
Dbuffer_assignment_test.cc236 builder.AddInstruction(HloInstruction::CreateUnary(r0f32_, opcode, param)); in BuildR0F32UnaryOpComputation()
388 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kNegate, param0)); in TEST_F()
413 HloInstruction::CreateUnary(const0->shape(), HloOpcode::kCopy, const0)); in TEST_F()
482 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kNegate, param)); in TEST_F()
484 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kNegate, neg_1)); in TEST_F()
807 HloInstruction::CreateUnary(f32a100x10_, HloOpcode::kExp, param0)); in TEST_F()
809 HloInstruction::CreateUnary(f32a100x10_, HloOpcode::kExp, exp1)); in TEST_F()
818 HloInstruction::CreateUnary(f32vec10_, HloOpcode::kExp, reduce)); in TEST_F()
971 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kExp, param0)); in TEST_F()
973 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kTanh, exp1)); in TEST_F()
[all …]
Dhlo_computation_test.cc56 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, param)); in CreateNegateComputation()
141 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, constant)); in TEST_F()
143 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, negate1)); in TEST_F()
156 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, constant)); in TEST_F()
160 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, negate1)); in TEST_F()
424 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, constant)); in TEST_F()
450 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, constant)); in TEST_F()
454 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, constant)); in TEST_F()
482 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, param)); in TEST_F()
Dhlo_dataflow_analysis_test.cc88 HloInstruction::CreateUnary(scalar_shape_, opcode, param0)); in CreateR0F32UnaryOpComputation()
691 auto negate = outer_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_P()
1108 auto bitcast = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_P()
1147 HloInstruction::CreateUnary(tuple->shape(), HloOpcode::kCopy, tuple)); in TEST_P()
1238 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kNegate, param)); in TEST_P()
1240 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kExp, negate)); in TEST_P()
1242 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kLog, exp)); in TEST_P()
1276 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kNegate, param0)); in TEST_P()
1278 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kExp, param1)); in TEST_P()
1337 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kExp, constant)); in TEST_P()
[all …]
Dcopy_insertion_test.cc142 HloInstruction::CreateUnary(copy_shape, HloOpcode::kCopy, constant)); in TEST_F()
144 HloInstruction::CreateUnary(copy_shape, HloOpcode::kCopy, constant)); in TEST_F()
148 HloInstruction::CreateUnary(add->shape(), HloOpcode::kCopy, add)); in TEST_F()
238 HloInstruction* bitcast = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
261 HloInstruction* bitcast = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
282 HloInstruction* bitcast = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
955 auto iter_value = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
957 auto data_value = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
1319 cond_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
1517 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte0)); in TEST_F()
[all …]
Dhlo_alias_analysis_test.cc234 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte0)); in TEST_F()
236 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte1)); in TEST_F()
371 auto negate_1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
373 auto negate_2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
731 auto negate = outer_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
948 auto negate = body_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
1011 auto bitcast = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
1031 auto bitcast = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
1064 HloInstruction::CreateUnary(init->shape(), HloOpcode::kExp, body_param)); in TEST_F()
1071 HloInstruction::CreateUnary(init->shape(), HloOpcode::kNegate, init)); in TEST_F()
Dhlo_domain_test.cc395 infeed_data->parent()->AddInstruction(HloInstruction::CreateUnary( in TEST_F()
402 infeed_data->parent()->AddInstruction(HloInstruction::CreateUnary( in TEST_F()
638 HloInstruction::CreateUnary(tuple0->operand(1)->shape(), HloOpcode::kCopy, in TEST_F()
643 HloInstruction::CreateUnary(tuple0->shape(), HloOpcode::kCopy, tuple0)); in TEST_F()
651 HloInstruction::CreateUnary(gte1->shape(), HloOpcode::kCopy, gte1)); in TEST_F()
Dinstruction_fusion_test.cc173 HloInstruction::CreateUnary(shape, HloOpcode::kAbs, binary1)); in TEST_F()
358 HloInstruction::CreateUnary(shape, HloOpcode::kFloor, param0)); in TEST_F()
362 HloInstruction::CreateUnary(shape, HloOpcode::kAbs, unary1)); in TEST_F()
386 HloInstruction::CreateUnary(shape, HloOpcode::kAbs, binary1)); in TEST_F()
Dhlo_reachability_test.cc110 HloInstruction::CreateUnary(r0f32, HloOpcode::kNegate, constant2)); in TEST_F()
112 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, negate)); in TEST_F()
116 HloInstruction::CreateUnary(r0f32, HloOpcode::kCopy, exp)); in TEST_F()
Dhlo_ordering_test.cc135 auto negate = body_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
218 auto negate = body_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
417 HloInstruction::CreateUnary(scalar_shape, HloOpcode::kNegate, param)); in TEST_F()
465 HloInstruction::CreateUnary(scalar_shape, HloOpcode::kNegate, param)); in TEST_F()
Dlayout_assignment_test.cc178 HloInstruction::CreateUnary(ashape, HloOpcode::kNegate, add)); in TEST_F()
180 HloInstruction::CreateUnary(ashape, HloOpcode::kNegate, negate1)); in TEST_F()
228 auto negate = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
357 HloInstruction::CreateUnary(ashape, HloOpcode::kLog, param)); in TEST_F()
361 HloInstruction::CreateUnary(bshape, HloOpcode::kTanh, reshape)); in TEST_F()
399 HloInstruction::CreateUnary(ashape, HloOpcode::kLog, param)); in TEST_F()
403 HloInstruction::CreateUnary(bshape, HloOpcode::kTanh, transpose)); in TEST_F()
484 HloInstruction::CreateUnary(f32_34, HloOpcode::kTanh, broadcast)); in TEST_F()
825 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
Dhlo_schedule_test.cc194 HloInstruction* new_root = entry->AddInstruction(HloInstruction::CreateUnary( in TEST_F()
259 HloInstruction::CreateUnary(ShapeUtil::MakeShape(PRED, {}), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc56 HloInstruction* exp0 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
75 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
94 HloInstruction* exp0 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
96 HloInstruction* bitcast0 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
114 HloInstruction* exp0 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
136 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
155 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
318 HloInstruction::CreateUnary(param_shape, HloOpcode::kExp, param0)); in TEST_F()
322 HloInstruction::CreateUnary(result_shape, HloOpcode::kNegate, reshape2)); in TEST_F()
352 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcopy_test.cc45 builder.AddInstruction(HloInstruction::CreateUnary( in TestCopyOp()
97 HloInstruction::CreateUnary(shape, HloOpcode::kCopy, param0)); in XLA_TEST_F()
115 auto copy = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
118 HloInstruction::CreateUnary(copy->shape(), HloOpcode::kCopy, copy)); in XLA_TEST_F()
143 builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
175 builder.AddInstruction(HloInstruction::CreateUnary( in TestCopyConstantLayout021()
209 builder.AddInstruction(HloInstruction::CreateUnary( in TestCopyConstantLayoutR4()
Dfusion_test.cc106 root_hlo = HloInstruction::CreateUnary(answer_shape, opcode, hlos[1]); in TestElementwise2D()
217 auto negate7 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
253 auto copy1 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
489 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
507 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
525 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
546 auto negate3 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
565 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
583 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
636 auto negate3 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion_test.cc38 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
58 HloInstruction* negate1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
78 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
97 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_copy_test.cc43 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()

123