Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_intrinsic_test.cc136 HloOpcode::kTanh, kTriple_x86_64, "",
140 HloOpcode::kTanh, kTriple_x86_64, "+avx",
144 HloOpcode::kTanh, kTriple_android_arm, "",
/external/tensorflow/tensorflow/compiler/xla/service/
Dlogistic_expander.cc46 MakeUnaryHlo(HloOpcode::kTanh, in ExpandLogisticWithTanh()
81 case LogisticExpansionType::kTanh: in ExpandInstruction()
Dlogistic_expander.h28 kTanh, // Expands as 0.5 + 0.5*tanh(0.5*x) enumerator
Dlogistic_expander_test.cc65 LogisticExpander logistic_expander(LogisticExpansionType::kTanh); in TEST_F()
Dzero_sized_hlo_elimination_test.cc59 zero_sized_param_->shape(), HloOpcode::kTanh, zero_sized_param_)); in TEST_F()
Dmemory_space_assignment_test.cc261 HloInstruction::CreateUnary(shape, HloOpcode::kTanh, p0)); in CreateEvictAndPrefetchModule()
583 HloInstruction::CreateUnary(shape, HloOpcode::kTanh, p0)); in TEST_P()
652 HloInstruction::CreateUnary(shape, HloOpcode::kTanh, p0)); in TEST_P()
1423 a.buffer->defining_instruction()->opcode() == HloOpcode::kTanh; in TEST_P()
1425 b.buffer->defining_instruction()->opcode() == HloOpcode::kTanh; in TEST_P()
3622 HloInstruction::CreateUnary(shape, HloOpcode::kTanh, p0)); in TEST_P()
3626 HloInstruction::CreateUnary(shape, HloOpcode::kTanh, tanh0)); in TEST_P()
3630 HloInstruction::CreateUnary(shape, HloOpcode::kTanh, tanh1)); in TEST_P()
3634 HloInstruction::CreateUnary(shape, HloOpcode::kTanh, tanh2)); in TEST_P()
3638 HloInstruction::CreateUnary(shape, HloOpcode::kTanh, tanh3)); in TEST_P()
[all …]
Dhlo_opcode.h148 V(kTanh, "tanh", 1) \
Dlayout_assignment_test.cc361 HloInstruction::CreateUnary(bshape, HloOpcode::kTanh, reshape)); in TEST_F()
403 HloInstruction::CreateUnary(bshape, HloOpcode::kTanh, transpose)); in TEST_F()
484 HloInstruction::CreateUnary(f32_34, HloOpcode::kTanh, broadcast)); in TEST_F()
Dinstruction_fusion.cc185 case HloOpcode::kTanh: in IsExpensive()
Dhlo_instruction.cc914 case HloOpcode::kTanh: in CreateUnary()
1740 case HloOpcode::kTanh: in CloneWithNewOperands()
2172 case HloOpcode::kTanh: in IdenticalSlowPath()
2639 case HloOpcode::kTanh: in IsOpElementwise()
3174 case HloOpcode::kTanh: in Visit()
Dhlo_rematerialization_test.cc279 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kTanh, rng)); in TEST_F()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dop_version.cc185 {{OperatorType::kTanh, 1}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
186 {{OperatorType::kTanh, 2}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
187 {{OperatorType::kTanh, 3}, kPendingReleaseOpVersion}, in GetMinimumRuntimeVersionForModel()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtarget_util.h63 kTanh, enumerator
Dtarget_util.cc156 case TargetDeviceFunctionID::kTanh: { in GetDeviceFunctionRoot()
Delemental_ir_emitter.cc270 return EmitDeviceMathCall(TargetDeviceFunctionID::kTanh, {value}, in EmitTanh()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_lstm.cc158 if (!MatchOperatorInputs(*final_output_mul, *model, OperatorType::kTanh, in Run()
200 &state_remember_sig, OperatorType::kTanh, in Run()
Dreorder_elementwise_unary.cc41 case OperatorType::kTanh: in IsElementwiseOperator()
Dquantize.cc105 OperatorType::kTanh, in SupportsQuantization()
372 if (op.type == OperatorType::kTanh) { in ChooseHardcodedQuantizationForOperatorOutput()
Dhardcode_min_max.cc509 case OperatorType::kTanh: in Run()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.proto60 kTanh = 5; enumerator
Ddnn.cc107 case ActivationMode::kTanh: in ActivationModeString()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_exporter.cc74 return stream_executor::dnn::kTanh; in ConvertConvActivationMode()
Dhlo_utils.cc415 return xla::HloOpcode::kTanh; in MhloToHloOpcode()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h83 kTanh, enumerator
842 TanhOperator() : Operator(OperatorType::kTanh) {}
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc355 dynamic_slice_shape, HloOpcode::kTanh, dynamic_slice4)); in TEST_F()
362 {HloOpcode::kTanh, HloOpcode::kDynamicSlice, HloOpcode::kReshape, in TEST_F()

12