Home
last modified time | relevance | path

Searched refs:kTanh (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dzero_sized_hlo_elimination_test.cc60 zero_sized_param_->shape(), HloOpcode::kTanh, zero_sized_param_)); in TEST_F()
Dhlo_opcode.h126 V(kTanh, "tanh") \
Dlayout_assignment_test.cc332 HloInstruction::CreateUnary(bshape, HloOpcode::kTanh, reshape)); in TEST_F()
375 HloInstruction::CreateUnary(bshape, HloOpcode::kTanh, transpose)); in TEST_F()
456 HloInstruction::CreateUnary(f32_34, HloOpcode::kTanh, broadcast)); in TEST_F()
Dinstruction_fusion.cc121 case HloOpcode::kTanh: in IsExpensive()
Dhlo_cost_analysis.cc94 opcode == HloOpcode::kTanh || opcode == HloOpcode::kSin || in HandleElementwiseOp()
Dhlo_rematerialization_test.cc350 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kTanh, rng)); in TEST_F()
Dhlo_instruction.cc240 case HloOpcode::kTanh: in CreateUnary()
1224 case HloOpcode::kTanh: in CloneWithNewOperands()
1727 case HloOpcode::kTanh: in IdenticalSlowPath()
2614 case HloOpcode::kTanh: in Visit()
2966 case HloOpcode::kTanh: in IsElementwise()
Delemental_ir_emitter.cc565 case HloOpcode::kTanh: { in EmitComplexUnaryOp()
1312 case HloOpcode::kTanh: in MakeElementGenerator()
Dhlo_graph_dumper.cc920 case HloOpcode::kTanh: in GetInstructionColor()
Dbuffer_assignment_test.cc810 HloInstruction::CreateUnary(f32vec100_, HloOpcode::kTanh, exp1)); in TEST_F()
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Didentify_lstm.cc220 if (!MatchOperatorInputs(*final_output_mul, *model, OperatorType::kTanh, in Run()
262 &state_remember_sig, OperatorType::kTanh, in Run()
Dquantize.cc50 type == OperatorType::kTanh || type == OperatorType::kMul || in SupportsQuantization()
294 if (op.type == OperatorType::kTanh) { in ChooseHardcodedQuantizationForOperatorOutput()
Dhardcode_min_max.cc322 case OperatorType::kTanh: in Run()
Dpropagate_fixed_sizes.cc1355 case OperatorType::kTanh: in Run()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Delemental_ir_emitter.cc34 case HloOpcode::kTanh: { in EmitFloatUnaryOp()
Dcpu_instruction_fusion_test.cc289 dynamic_slice_shape, HloOpcode::kTanh, dynamic_slice4)); in TEST_F()
296 {HloOpcode::kTanh, HloOpcode::kDynamicSlice, HloOpcode::kBitcast, in TEST_F()
/external/tensorflow/tensorflow/contrib/lite/toco/
Dmodel.h69 kTanh, enumerator
579 TanhOperator() : Operator(OperatorType::kTanh) {}
Dtooling_util.cc1526 case OperatorType::kTanh: { in EstimateArithmeticOpsCount()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc263 case HloOpcode::kTanh: in EmitFloatUnaryOp()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc68 case ActivationMode::kTanh: in ActivationModeString()
Ddnn.h864 kTanh,
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
Doperator_test.cc108 CheckSimpleOperator<TanhOperator>("TANH", OperatorType::kTanh); in TEST_F()
Doperator.cc860 new SimpleOperator<TanhOperator>("TANH", OperatorType::kTanh)); in BuildOperatorList()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc843 case dnn::ActivationMode::kTanh: in ScopedActivationDescriptor()
3921 case dnn::ActivationMode::kTanh: in DoActivate()
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
Dhlo_parser.cc486 case HloOpcode::kTanh: { in ParseInstruction()

12