| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | elu_test.cc | 28 TEST(Elu, 4D) {
47 TEST(Elu, 3D) {
65 TEST(Elu, 2D) {
82 TEST(Elu, 1D) {
97 TEST(Elu, MultiThreading) { in TEST() argument
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EluGrad.pbtxt | 7 The backpropagated gradients to the corresponding Elu operation.
13 The outputs of the corresponding Elu operation.
23 summary: "Computes gradients for the exponential linear (Elu) operation."
|
| D | api_def_Elu.pbtxt | 2 graph_op_name: "Elu"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Elu.pbtxt | 2 name: "Elu"
24 name: "Elu"
46 name: "Elu"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Elu.pbtxt | 2 name: "Elu"
24 name: "Elu"
46 name: "Elu"
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_Elu.pbtxt | 2 graph_op_name: "Elu"
4 name: "nn.Elu"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | fused_kernel_matcher.mlir | 44 …lon = 0.000000e+00 : f32, explicit_paddings = [], fused_ops = ["BiasAdd", "Elu"], padding = "SAME"…
49 %2 = "tf.Elu"(%1) : (tensor<*xf32>) -> tensor<*xf32>
59 %2 = "tf.Elu"(%1) : (tensor<*xf32>) -> tensor<*xf32>
68 // CHECK-DAG: %[[VAL_0:.*]] = "tf.Elu"(%[[VAL]]) : (tensor<*xf32>) -> tensor<*xf32>
74 %2 = "tf.Elu"(%1) : (tensor<*xf32>) -> tensor<*xf32>
84 %2 = "tf.Elu"(%0) : (tensor<*xf32>) -> tensor<*xf32>
95 %2 = "tf.Elu"(%1) : (tensor<*xf32>) -> tensor<*xf32>
106 %2 = "tf.Elu"(%1) : (tensor<*xf32>) -> tensor<*xf32>
152 …rg1, %arg2, %arg0) {epsilon = 0.000000e+00 : f32, fused_ops = ["BiasAdd", "Elu"], transpose_a = fa…
157 %5 = "tf.Elu"(%4) : (tensor<*xf32>) -> tensor<*xf32>
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | elu_op.cc | 28 XlaOp Elu(XlaOp x) { in Elu() function
53 ctx->SetOutput(0, xla::Elu(ctx->Input(0))); in Compile()
|
| D | elu_op.h | 22 XlaOp Elu(XlaOp x);
|
| D | unary_ops_composition.cc | 75 ADD_XLA_OP_GENERATOR(Elu); in PopulateXlaOpGeneratorMap()
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_Elu.pbtxt | 2 graph_op_name: "Elu"
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | elu.h | 26 inline void Elu(const RuntimeShape& input_shape, const float* input_data, in Elu() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unary_ops_composition.cc | 195 auto elu = functor::Elu<Eigen::DefaultDevice, T>(); \
252 REGISTER_COMPUTE_FN(Elu); in UnaryOpsCompositionSupport()
317 REGISTER_COMPUTE_FN(Elu); in UnaryOpsCompositionSupport()
381 REGISTER_COMPUTE_FN(Elu); in UnaryOpsCompositionSupport()
|
| D | relu_op.cc | 117 void Elu<GPUDevice, T>::operator()(const GPUDevice& d, \
120 extern template struct Elu<GPUDevice, T>; \
|
| D | fused_eigen_output_kernels.h | 103 struct Elu { struct
355 using WithBiasAddAndElu = BiasAddOutputKernel<T, Elu>;
365 using WithFusedBatchNormAndElu = FusedBatchNormOutputKernel<T, Elu>;
|
| D | relu_op_functor.h | 140 struct Elu { struct
|
| D | relu_op_gpu.cu.cc | 248 template struct functor::Elu<GPUDevice, T>; \
|
| D | relu_op.h | 201 functor::Elu<Device, T> functor; in Operate()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | mkl_remapper_test.cc | 84 ops::Identity(fetch, ops::Elu(activate, addop)); in FuseConv2DWithBiasAndAddNOrAdd()
194 CREATE_CONV2DFUSION_TEST(data_format, addop, Elu, bcast); \
248 return ops::Identity(fetch, ops::Elu(activate, bias_add)); \
|
| D | remapper_test.cc | 574 return ops::Identity(fetch, ops::Elu(activate, bias_add)); in TEST_F()
672 return ops::Identity(fetch, ops::Elu(activate, bias_add)); in RunTest()
868 return ops::Identity(fetch, ops::Elu(activate, batch_norm.y)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_relu_op_test.cc | 127 TEST_ALL_SIZES(Elu)
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | elu.cc | 122 optimized_ops::Elu(GetTensorShape(input), GetTensorData<float>(input), in EluEval()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad_test.cc | 33 using ops::Elu;
193 auto y = Elu(scope_, x); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | elementwise_test.cc | 45 TEST_F(OpenCLOperationTest, Elu) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/resources/ |
| D | decomposition_lib.mlir | 63 // This is a wong decomposition and used to verify that tf.Elu isn't decomposed
|