Home
last modified time | relevance | path

Searched refs:kExp (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc50 ShapeUtil::MakeShape(S32, {1024, 256}), HloOpcode::kExp, arg0)); in TEST_F()
69 ShapeUtil::MakeShape(S32, {256, 1024}), HloOpcode::kExp, arg1)); in TEST_F()
88 ShapeUtil::MakeShape(S32, {2, 512, 2, 128}), HloOpcode::kExp, arg0)); in TEST_F()
109 ShapeUtil::MakeShape(S32, {2, 512, 2, 128}), HloOpcode::kExp, arg0)); in TEST_F()
131 ShapeUtil::MakeShape(S32, {256, 32 * 1024}), HloOpcode::kExp, arg1)); in TEST_F()
150 ShapeUtil::MakeShape(S32, {256, 1024}), HloOpcode::kExp, arg1)); in TEST_F()
169 ShapeUtil::MakeShape(S32, {1024, 256}), HloOpcode::kExp, arg1)); in TEST_F()
256 HloInstruction::CreateUnary(param_shape, HloOpcode::kExp, param0)); in TEST_F()
266 module.get(), {HloOpcode::kNegate, HloOpcode::kBitcast, HloOpcode::kExp, in TEST_F()
347 HloInstruction::CreateUnary(param_shape, HloOpcode::kExp, param0)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cse_test.cc212 constant->shape(), HloOpcode::kExp, constant)); in TEST_F()
214 constant->shape(), HloOpcode::kExp, constant)); in TEST_F()
216 constant->shape(), HloOpcode::kExp, constant)); in TEST_F()
243 constant->shape(), HloOpcode::kExp, constant)); in TEST_F()
247 constant->shape(), HloOpcode::kExp, constant)); in TEST_F()
274 constant->shape(), HloOpcode::kExp, constant)); in TEST_F()
278 constant->shape(), HloOpcode::kExp, constant)); in TEST_F()
354 constant->shape(), HloOpcode::kExp, constant)); in TEST_F()
361 constant->shape(), HloOpcode::kExp, constant)); in TEST_F()
Dhlo_instruction_test.cc184 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
186 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
477 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
545 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
607 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, foo)); in TEST_F()
629 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, constant)); in TEST_F()
664 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, constant)); in TEST_F()
666 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, exp1)); in TEST_F()
668 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, exp2)); in TEST_F()
685 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, constant)); in TEST_F()
[all …]
Dbuffer_liveness_test.cc114 HloInstruction::CreateUnary(vec_, HloOpcode::kExp, negate)); in TEST_F()
161 HloInstruction::CreateUnary(vec_, HloOpcode::kExp, param1)); in TEST_F()
208 HloInstruction::CreateUnary(vec_, HloOpcode::kExp, param)); in TEST_F()
244 HloInstruction::CreateUnary(vec_, HloOpcode::kExp, param)); in TEST_F()
286 HloInstruction::CreateUnary(vec_, HloOpcode::kExp, param)); in TEST_F()
370 HloInstruction::CreateUnary(vec_, HloOpcode::kExp, negate)); in TEST_F()
Dalgebraic_simplifier.cc554 if (lhs->opcode() == HloOpcode::kExp && rhs->opcode() == HloOpcode::kExp) { in HandleDivide()
561 divide, HloInstruction::CreateUnary(divide->shape(), HloOpcode::kExp, in HandleDivide()
566 if (rhs->opcode() == HloOpcode::kExp) { in HandleDivide()
572 HloInstruction::CreateUnary(divide->shape(), HloOpcode::kExp, negate)); in HandleDivide()
988 if (lhs->opcode() == HloOpcode::kExp && rhs->opcode() == HloOpcode::kExp) { in HandleMultiply()
994 HloInstruction::CreateUnary(multiply->shape(), HloOpcode::kExp, add)); in HandleMultiply()
1003 if (operand->opcode() == HloOpcode::kExp && in HandleLog()
1363 if (lhs->opcode() == HloOpcode::kExp) { in HandlePower()
1367 power, HloInstruction::CreateUnary(power->shape(), HloOpcode::kExp, in HandlePower()
Dhlo_opcode.h75 V(kExp, "exponential") \
Dhlo_dataflow_analysis_test.cc1215 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kExp, negate)); in TEST_P()
1252 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kExp, param1)); in TEST_P()
1309 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kExp, constant)); in TEST_P()
1395 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kExp, param)); in TEST_P()
1428 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kExp, param)); in TEST_P()
1467 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kExp, param)); in TEST_P()
1525 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kExp, param)); in TEST_P()
Dliveness_util_test.cc130 HloInstruction::CreateUnary(shape, HloOpcode::kExp, param)); in TEST_F()
179 HloInstruction::CreateUnary(shape, HloOpcode::kExp, param)); in TEST_F()
Dbuffer_assignment_test.cc644 HloInstruction::CreateUnary(f32a100x10_, HloOpcode::kExp, param0)); in TEST_F()
646 HloInstruction::CreateUnary(f32a100x10_, HloOpcode::kExp, exp1)); in TEST_F()
655 HloInstruction::CreateUnary(f32vec10_, HloOpcode::kExp, reduce)); in TEST_F()
808 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kExp, param0)); in TEST_F()
812 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kExp, tanh)); in TEST_F()
1052 HloInstruction::CreateUnary(vec_shape, HloOpcode::kExp, call_param)); in TEST_F()
Dinstruction_fusion.cc102 case HloOpcode::kExp: in IsExpensive()
Dalgebraic_simplifier_test.cc309 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, param1)); in TEST_F()
608 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, param0)); in TEST_F()
610 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, param1)); in TEST_F()
636 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, param0)); in TEST_F()
638 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, param1)); in TEST_F()
664 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, param0)); in TEST_F()
714 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, param0)); in TEST_F()
738 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, param0)); in TEST_F()
740 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, param1)); in TEST_F()
Dhlo_ordering_test.cc55 HloInstruction::CreateUnary(vec, HloOpcode::kExp, param)); in TEST_F()
Dhlo_cost_analysis_test.cc364 HloInstruction::CreateUnary(r2f32, HloOpcode::kExp, add)); in TEST_F()
Dhlo_cost_analysis.cc93 if (opcode == HloOpcode::kExp || opcode == HloOpcode::kPower || in HandleElementwiseOp()
Dhlo_rematerialization_test.cc352 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kExp, rng)); in TEST_F()
Dhlo_instruction.cc229 case HloOpcode::kExp: in CreateUnary()
1213 case HloOpcode::kExp: in CloneWithNewOperands()
1700 case HloOpcode::kExp: in IdenticalSlowPath()
2606 case HloOpcode::kExp: in Visit()
2955 case HloOpcode::kExp: in IsElementwise()
Delemental_ir_emitter.cc406 case HloOpcode::kExp: in EmitFloatUnaryOp()
502 case HloOpcode::kExp: { in EmitComplexUnaryOp()
1302 case HloOpcode::kExp: in MakeElementGenerator()
Dhlo_computation_test.cc478 HloInstruction::CreateUnary(r0f32_, HloOpcode::kExp, negate)); in TEST_F()
Dcopy_insertion_test.cc945 iter_param->shape(), HloOpcode::kExp, iter_param)); in TEST_F()
947 data_param->shape(), HloOpcode::kExp, data_param)); in TEST_F()
Dhlo_alias_analysis_test.cc861 HloInstruction::CreateUnary(init->shape(), HloOpcode::kExp, body_param)); in TEST_F()
Dhlo_graph_dumper.cc891 case HloOpcode::kExp: in GetInstructionColor()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion_test.cc34 ShapeUtil::MakeShape(S32, {}), HloOpcode::kExp, const0)); in TEST_F()
74 ShapeUtil::MakeShape(S32, {}), HloOpcode::kExp, const0)); in TEST_F()
93 ShapeUtil::MakeShape(S32, {}), HloOpcode::kExp, const0)); in TEST_F()
/external/tensorflow/tensorflow/contrib/lite/toco/
Dmodel.h47 kExp, enumerator
865 ExpOperator() : Operator(OperatorType::kExp) {}
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
Doperator_test.cc109 CheckSimpleOperator<ExpOperator>("EXP", OperatorType::kExp); in TEST_F()
Doperator.cc861 ops.emplace_back(new SimpleOperator<ExpOperator>("EXP", OperatorType::kExp)); in BuildOperatorList()

12