Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc64 HloInstruction::CreateUnary(vshape, HloOpcode::kNegate, add1)); in TEST_F()
98 HloInstruction::CreateUnary(vshape, HloOpcode::kNegate, input)); in TEST_F()
100 HloInstruction::CreateUnary(vshape, HloOpcode::kCeil, negate)); in TEST_F()
102 HloInstruction::CreateUnary(vshape, HloOpcode::kExp, ceil)); in TEST_F()
104 HloInstruction::CreateUnary(vshape, HloOpcode::kFloor, exp)); in TEST_F()
148 HloInstruction::CreateUnary(vshape, HloOpcode::kNegate, input)); in TEST_F()
150 HloInstruction::CreateUnary(vshape, HloOpcode::kCeil, negate)); in TEST_F()
178 HloInstruction::CreateUnary(cshape, HloOpcode::kExp, reduce)); in TEST_F()
180 HloInstruction::CreateUnary(cshape, HloOpcode::kFloor, exp)); in TEST_F()
245 HloInstruction::CreateUnary(vshape, HloOpcode::kNegate, constant)); in TEST_F()
[all …]
Dcpu_dyn_shape_test.cc36 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_test.cc293 HloInstruction::CreateUnary(shape, HloOpcode::kTanh, p0)); in CreateEvictAndPrefetchModule()
511 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, p0)); in TEST_P()
513 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate0)); in TEST_P()
515 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate1)); in TEST_P()
517 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate2)); in TEST_P()
519 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate3)); in TEST_P()
521 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate4)); in TEST_P()
523 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate5)); in TEST_P()
618 HloInstruction::CreateUnary(shape, HloOpcode::kTanh, p0)); in TEST_P()
687 HloInstruction::CreateUnary(shape, HloOpcode::kTanh, p0)); in TEST_P()
[all …]
Dhlo_memory_scheduler_test.cc76 HloInstruction::CreateUnary(vec, HloOpcode::kAbs, param)); in TEST_F()
78 HloInstruction::CreateUnary(vec, HloOpcode::kExp, param)); in TEST_F()
83 HloInstruction::CreateUnary(vec, HloOpcode::kNegate, exp)); in TEST_F()
225 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, lit)); in TEST_F()
228 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, abs_const)); in TEST_F()
235 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, abs_const)); in TEST_F()
282 HloInstruction::CreateUnary(r1f32, HloOpcode::kExp, c3)); in TEST_F()
Dhlo_instruction_test.cc190 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
192 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
303 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, c0)); in TEST_F()
311 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, addtotal)); in TEST_F()
483 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
485 HloInstruction::CreateUnary(r0f32_, HloOpcode::kLog, foo)); in TEST_F()
551 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
613 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
615 HloInstruction::CreateUnary(r0f32_, HloOpcode::kLog, foo)); in TEST_F()
635 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, constant)); in TEST_F()
[all …]
Dhlo_cse_test.cc196 auto exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
198 auto exp2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
200 auto exp3 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
409 auto exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
413 auto exp2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
440 auto exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
444 auto exp2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
518 auto negate1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
520 auto exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
525 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
[all …]
Dhlo_dce_test.cc140 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
144 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
169 auto dead_negate = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
177 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
221 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, param)); in TEST_F()
233 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, param)); in TEST_F()
290 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, param)); in TEST_F()
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()
Dasync_collective_creator.cc69 std::unique_ptr<HloInstruction> done = HloInstruction::CreateUnary( in Run()
99 std::unique_ptr<HloInstruction> done = HloInstruction::CreateUnary( in Run()
143 computation->AddInstruction(HloInstruction::CreateUnary( in Run()
Dhlo_dataflow_analysis_test.cc98 HloInstruction::CreateUnary(scalar_shape_, opcode, param0)); in CreateR0F32UnaryOpComputation()
611 auto negate = level2_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_P()
699 auto negate = outer_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_P()
979 HloInstruction::CreateUnary(tuple->shape(), HloOpcode::kCopy, tuple)); in TEST_P()
1014 auto barrier = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_P()
1049 auto copy_done = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_P()
1273 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kNegate, param)); in TEST_P()
1275 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kExp, negate)); in TEST_P()
1277 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kLog, exp)); in TEST_P()
1311 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kNegate, param0)); in TEST_P()
[all …]
Dhlo_rematerialization_test.cc279 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kTanh, rng)); in TEST_F()
281 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kExp, rng)); in TEST_F()
433 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kCopy, param)); in TEST_F()
436 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kNegate, copy)); in TEST_F()
438 auto negate_a_2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
442 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kNegate, copy)); in TEST_F()
444 auto negate_b_2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
534 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kNegate, gte)); in TEST_P()
Dbuffer_assignment_test.cc262 builder.AddInstruction(HloInstruction::CreateUnary(r0f32_, opcode, param)); in BuildR0F32UnaryOpComputation()
412 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kNegate, param0)); in TEST_F()
437 HloInstruction::CreateUnary(const0->shape(), HloOpcode::kCopy, const0)); in TEST_F()
506 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kNegate, param)); in TEST_F()
508 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kNegate, neg_1)); in TEST_F()
843 HloInstruction::CreateUnary(f32vec10_color1, HloOpcode::kNegate, data)); in TEST_F()
998 HloInstruction::CreateUnary(f32a100x10_, HloOpcode::kExp, param0)); in TEST_F()
1000 HloInstruction::CreateUnary(f32a100x10_, HloOpcode::kExp, exp1)); in TEST_F()
1009 HloInstruction::CreateUnary(f32vec10_, HloOpcode::kExp, reduce)); in TEST_F()
1162 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kExp, param0)); in TEST_F()
[all …]
Dhlo_computation_test.cc55 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, param)); in CreateNegateComputation()
140 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, constant)); in TEST_F()
142 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, negate1)); in TEST_F()
405 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, constant)); in TEST_F()
432 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, constant)); in TEST_F()
436 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, constant)); in TEST_F()
464 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, param)); in TEST_F()
Dhlo_reachability_test.cc111 HloInstruction::CreateUnary(r0f32, HloOpcode::kNegate, constant2)); in TEST_F()
113 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, negate)); in TEST_F()
117 HloInstruction::CreateUnary(r0f32, HloOpcode::kCopy, exp)); in TEST_F()
Dhlo_domain_test.cc488 infeed_data->parent()->AddInstruction(HloInstruction::CreateUnary( in TEST_F()
495 infeed_data->parent()->AddInstruction(HloInstruction::CreateUnary( in TEST_F()
731 HloInstruction::CreateUnary(tuple0->operand(1)->shape(), HloOpcode::kCopy, in TEST_F()
736 HloInstruction::CreateUnary(tuple0->shape(), HloOpcode::kCopy, tuple0)); in TEST_F()
744 HloInstruction::CreateUnary(gte1->shape(), HloOpcode::kCopy, gte1)); in TEST_F()
Dinstruction_fusion_test.cc126 HloInstruction::CreateUnary(shape, HloOpcode::kAbs, binary1)); in TEST_F()
319 HloInstruction::CreateUnary(shape, HloOpcode::kFloor, param0)); in TEST_F()
325 HloInstruction::CreateUnary(shape, HloOpcode::kAbs, unary1)); in TEST_F()
353 HloInstruction::CreateUnary(shape, HloOpcode::kAbs, binary1)); in TEST_F()
Dhlo_ordering_test.cc134 auto negate = body_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
217 auto negate = body_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
424 HloInstruction::CreateUnary(scalar_shape, HloOpcode::kNegate, param)); in TEST_F()
472 HloInstruction::CreateUnary(scalar_shape, HloOpcode::kNegate, param)); in TEST_F()
Dcopy_insertion_test.cc144 HloInstruction::CreateUnary(copy_shape, HloOpcode::kCopy, constant)); in TEST_F()
146 HloInstruction::CreateUnary(copy_shape, HloOpcode::kCopy, constant)); in TEST_F()
150 HloInstruction::CreateUnary(add->shape(), HloOpcode::kCopy, add)); in TEST_F()
1066 HloInstruction::CreateUnary(convert->shape(), HloOpcode::kExp, convert)); in TEST_F()
1069 auto data_value = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
1375 cond_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
1568 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte0)); in TEST_F()
1571 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte1)); in TEST_F()
1606 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte0)); in TEST_F()
1609 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte1)); in TEST_F()
[all …]
Dhlo_alias_analysis_test.cc267 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte0)); in TEST_F()
269 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte1)); in TEST_F()
398 auto negate_1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
400 auto negate_2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
753 auto negate = outer_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
911 HloInstruction::CreateUnary(shape, HloOpcode::kCopy, param0)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc57 HloInstruction* exp0 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
76 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
95 HloInstruction* exp0 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
97 HloInstruction* bitcast0 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
116 HloInstruction* exp0 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
138 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
157 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
302 HloInstruction::CreateUnary(param_shape, HloOpcode::kExp, param0)); in TEST_F()
306 HloInstruction::CreateUnary(result_shape, HloOpcode::kNegate, reshape2)); in TEST_F()
336 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcopy_test.cc43 builder.AddInstruction(HloInstruction::CreateUnary( in TestCopyOp()
95 HloInstruction::CreateUnary(shape, HloOpcode::kCopy, param0)); in XLA_TEST_F()
113 auto copy = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
116 HloInstruction::CreateUnary(copy->shape(), HloOpcode::kCopy, copy)); in XLA_TEST_F()
141 builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
173 builder.AddInstruction(HloInstruction::CreateUnary( in TestCopyConstantLayout021()
207 builder.AddInstruction(HloInstruction::CreateUnary( in TestCopyConstantLayoutR4()
Dcpu_gpu_fusion_test.cc103 root_hlo = HloInstruction::CreateUnary(answer_shape, opcode, hlos[1]); in TestElementwise2D()
214 auto negate7 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
250 auto copy1 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
486 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
504 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
522 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
543 auto negate3 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
562 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
580 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
632 auto negate3 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion_test.cc39 HloInstruction* log1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
59 HloInstruction* negate1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
79 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
98 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
120 auto log = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_dyn_shape_test.cc33 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
Dgpu_copy_test.cc42 builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()

123