Home
last modified time | relevance | path

Searched refs:Tanh (Results 1 – 25 of 175) sorted by relevance

1234567

/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
Dcwise_op_tanh_benchmark.cc22 %0 = "tf.Tanh"(%arg0): (tensor<?xf32>) -> tensor<?xf32>
32 BM_TFMlir(Tanh, mlir_input, "tanh_1d", 1, f32, 1.0, 0.0, /* num_threads */ 0)
38 BM_EigenScalar(Tanh, EXPR_BUILDER, 1, f32, 1.0, 0.0, /* num_threads */ 0)
44 BM_EigenVectorized(Tanh, EXPR_BUILDER, 1, f32, 1.0, 0.0, /* num_threads */ 0)
/external/tensorflow/tensorflow/compiler/jit/tests/
Dopens2s_gnmt_mixed_precision.golden_summary209 Tanh 17
254 Tanh 2
295 Tanh 2
327 Tanh 2
341 Tanh 2
377 Tanh 2
391 Tanh 2
405 Tanh 2
419 Tanh 2
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dgpu_op_tanh.cc21 GENERATE_AND_REGISTER_UNARY_GPU_KERNEL(Tanh, DT_HALF);
22 GENERATE_AND_REGISTER_UNARY_GPU_KERNEL(Tanh, DT_FLOAT);
23 GENERATE_AND_REGISTER_UNARY_GPU_KERNEL(Tanh, DT_DOUBLE);
/external/tensorflow/tensorflow/compiler/xla/python/
Dmlir.h21 GENERATE_AND_REGISTER_UNARY_CPU_KERNEL(Tanh, DT_HALF);
22 GENERATE_AND_REGISTER_UNARY_CPU_KERNEL(Tanh, DT_FLOAT);
23 GENERATE_AND_REGISTER_UNARY_CPU_KERNEL(Tanh, DT_DOUBLE);
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dtanh.h29 inline void Tanh(const RuntimeShape& input_shape, const float* input_data, in Tanh() function
42 inline void Tanh(const TanhParams&, const RuntimeShape& input_shape, in Tanh() function
46 Tanh(input_shape, input_data, output_shape, output_data); in Tanh()
49 inline void Tanh(const TanhParams& params, const RuntimeShape& input_shape, in Tanh() function
83 inline void Tanh(const TanhParams& params, const RuntimeShape& input_shape, in Tanh() function
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/
Dauto-fusion.mlir1 …-to-tfrt-pipeline="enable-native-ops=false auto-fusion-oplist=tf.Rsqrt,tf.Tanh auto-fusion-min-clu…
35 // CHECK-NOT: Tanh
37 %1 = "tf.Tanh"(%0) {T = f32, device="/device:CPU:0"} : (tensor<?xf32>) -> tensor<?xf32>
48 // CHECK: %[[RES1:.*]] = "tf.Tanh"(%[[RES0]])
60 %2 = "tf.Tanh"(%1) {T = f32, device="/device:CPU:0"} : (tensor<?xf32>) -> tensor<?xf32>
67 // CHECK: tf.Tanh
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor_test.cc86 op::Exp(op::Negate(op::Tanh(op::Parameter(0))))); in TEST_F()
106 op::Add(op::Negate(op::Tanh(op::Parameter(0))), in TEST_F()
107 op::Exp(op::Negate(op::Tanh(op::Parameter(0)))))); in TEST_F()
136 op::Add(op::Tanh(op::Parameter(0)), op::Constant())); in TEST_F()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTanh.pbtxt2 name: "Tanh"
26 name: "Tanh"
51 name: "Tanh"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DTanh.pbtxt2 name: "Tanh"
26 name: "Tanh"
51 name: "Tanh"
/external/tensorflow/tensorflow/compiler/jit/
Dintroduce_floating_point_jitter_pass_test.cc46 Output tanh_a = ops::Tanh(root.WithOpName("tanh_a"), sigmoid_a); in TEST()
47 Output tanh_b = ops::Tanh(root.WithOpName("tanh_b"), sigmoid_b); in TEST()
125 Output tanh = ops::Tanh(root.WithOpName("tanh"), sigmoid); in TEST()
155 Output tanh_s = ops::Tanh(root.WithOpName("tanh_s"), svd.s); in TEST()
156 Output tanh_u = ops::Tanh(root.WithOpName("tanh_u"), svd.u); in TEST()
157 Output tanh_v = ops::Tanh(root.WithOpName("tanh_v"), svd.v); in TEST()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlayout_optimization_move_transposes_begin.mlir8 // CHECK: %[[TANH:[0-9]*]] = "tf.Tanh"(%[[ARG_TRANSPOSE]]) {{.*}} tensor<1x8x4x4xf32>
11 %0 = "tf.Tanh"(%arg0) : (tensor<1x4x4x8xf32>) -> tensor<1x4x4x8xf32>
23 // CHECK: %[[TANH:[0-9]*]] = "tf.Tanh"(%[[ARG_TRANSPOSE]]) {{.*}} tensor<1x8x4x4xf32>
27 %0 = "tf.Tanh"(%arg0) : (tensor<1x4x4x8xf32>) -> tensor<1x4x4x8xf32>
57 // CHECK: %[[TANH:[0-9]*]] = "tf.Tanh"(%[[ARG_TRANSPOSE]]) {{.*}} tensor<1x8x4x4xf32>
61 %0 = "tf.Tanh"(%arg0) : (tensor<1x4x4x8xf32>) -> tensor<1x4x4x8xf32>
/external/tensorflow/tensorflow/core/transforms/remapper/tests/
Dcontraction.mlir43 …%[[PLACEHOLDER]], %[[WEIGHT]], %[[BIAS]]) {{.*}} name("Tanh") {{.*}} fused_ops = ["BiasAdd", "Tanh…
44 …%Tanh, %ctl_5 = Tanh(%BiasAdd) device("/device:CPU:0") name("Tanh") {T = f32} : (tensor<*xf32>) ->…
Donednn_contraction.mlir81 …%[[PLACEHOLDER]], %[[WEIGHT]], %[[BIAS]]) {{.*}} name("Tanh") {{.*}} fused_ops = ["BiasAdd", "Tanh…
82 …%Tanh, %ctl_5 = Tanh(%BiasAdd) device("/device:CPU:0") name("Tanh") {T = f32} : (tensor<*xf32>) ->…
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_Tanh.pbtxt2 graph_op_name: "Tanh"
4 name: "math.Tanh"
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/
Djit_i64_indexed_for_large_tensors.mlir22 // CHECK-SAME: %0 = \22tf.Tanh\22(%arg0)
38 // CHECK: %4 = "tf.Tanh"(%arg0)
42 %0 = "tf.Tanh"(%arg) : (tensor<*xf32>) -> tensor<*xf32>
Dtf_to_jit_invocations.mlir15 // CHECK-SAME: %0 = \22tf.Tanh\22(%arg0)
29 %0 = "tf.Tanh"(%arg) : (tensor<*xf32>) -> tensor<*xf32>
/external/tensorflow/tensorflow/compiler/xla/tests/
Dexhaustive_unary_test_complex.cc125 UNARY_TEST_COMPLEX_64(Tanh, {
137 Tanh,
220 UNARY_TEST_COMPLEX_128(Tanh, {
223 Tanh, +[](complex128 x) { return std::tanh(x); }); in __anone74288180e02()
Dexhaustive_unary_test_f64.cc96 UNARY_TEST_FLOAT_64(Tanh, {
105 Run(Tanh, std::tanh, error_spec_gen);
/external/skia/tests/sksl/intrinsics/
DTanh.asm.frag104 %27 = OpExtInst %float %1 Tanh %33
115 %41 = OpExtInst %v2float %1 Tanh %44
130 %54 = OpExtInst %v3float %1 Tanh %57
144 %68 = OpExtInst %v4float %1 Tanh %70
/external/deqp/external/vulkancts/data/vulkan/amber/graphicsfuzz/
Dcov-apfloat-tanh.amber92 %18 = OpExtInst %7 %1 Tanh %17
101 %27 = OpExtInst %7 %1 Tanh %26
/external/tensorflow/tensorflow/core/kernels/
Dunary_ops_composition.cc249 REGISTER_COMPUTE_FN(Tanh); in UnaryOpsCompositionSupport()
288 REGISTER_COMPUTE_FN_HELPER(Tanh, functor::tanh<T>);
315 REGISTER_COMPUTE_FN(Tanh); in UnaryOpsCompositionSupport()
343 REGISTER_COMPUTE_FN_HELPER(Tanh, functor::tanh<T>);
379 REGISTER_COMPUTE_FN(Tanh); in UnaryOpsCompositionSupport()
417 REGISTER_COMPUTE_FN_HELPER(Tanh, functor::tanh<T>);
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dtanh.h27 inline void Tanh(int32_t input_zero_point, int32_t input_range_radius, in Tanh() function
63 inline void Tanh(int32_t input_multiplier, int32_t input_left_shift, in Tanh() function
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/jit/
Dtf_jitrt_stream_analysis.mlir42 %3 = "tf.Tanh"(%2) : (tensor<?x?xf32>) -> tensor<?x?xf32>
66 %3 = "tf.Tanh"(%2) : (tensor<?x?xf32>) -> tensor<?x?xf32>
120 // (1 [Const] + 130 [Mul] + 3 [Sub] + 130 [BiasAdd] + 2 [Tanh]
138 // + 2 [Tanh] + 3 [Mul] + 130 [AddV2]) (cost @dyn_override_m)
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_Tanh.pbtxt2 graph_op_name: "Tanh"
/external/tensorflow/tensorflow/core/kernels/mlir_generated/op_definitions/
Dtanh.mlir.tmpl3 %0 = "tf.Tanh"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>

1234567