Home
last modified time | relevance | path

Searched refs:LeakyRelu (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_LeakyReluGrad.pbtxt7 The backpropagated gradients to the corresponding LeakyRelu operation.
13 The features passed as input to the corresponding LeakyRelu operation,
23 summary: "Computes rectified linear gradients for a LeakyRelu operation."
Dapi_def_LeakyRelu.pbtxt2 graph_op_name: "LeakyRelu"
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_LeakyRelu.pbtxt2 graph_op_name: "LeakyRelu"
4 name: "nn.LeakyRelu"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DLeakyRelu.pbtxt2 name: "LeakyRelu"
34 name: "LeakyRelu"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dconstant-fold.mlir38 %no = "tf.LeakyRelu"(%arg0) {alpha = 0.2 : f32} : (tensor<16xf32>) -> tensor<16xf32>
39 %0 = "tf.LeakyRelu"(%pos) {alpha = 0.3 : f32} : (tensor<f32>) -> tensor<f32>
40 %1 = "tf.LeakyRelu"(%neg) {alpha = 0.2 : f32} : (tensor<f32>) -> tensor<f32>
41 %2 = "tf.LeakyRelu"(%arg0) {alpha = 3.0 : f32} : (tensor<16xf32>) -> tensor<16xf32>
44 …// CHECK: [[NC1:%.*]] = "tf.LeakyRelu"(%arg0) {alpha = 2.000000e-01 : f32} : (tensor<16xf32>) -> t…
45 …// CHECK: [[NC2:%.*]] = "tf.LeakyRelu"(%arg0) {alpha = 3.000000e+00 : f32} : (tensor<16xf32>) -> t…
Dcanonicalize.mlir39 %2 = "tf.LeakyRelu"(%arg0) {alpha = 1.0 : f32} : (tensor<16xf32>) -> tensor<16xf32>
Dtf-ops.mlir194 // Test valid tf.LeakyRelu
198 %0 = "tf.LeakyRelu"(%arg0) {alpha = 0.2 : f32} : (tensor<16xf32>) -> tensor<16xf32>
206 %0 = "tf.LeakyRelu"(%arg0) {alpha = 1: i32}: (tensor<16xf32>) -> tensor<16xf32>
/external/tensorflow/tensorflow/core/kernels/
Drelu_op.cc108 void LeakyRelu<GPUDevice, T>::operator()( \
111 extern template struct LeakyRelu<GPUDevice, T>; \
Drelu_op_gpu.cu.cc173 template struct functor::LeakyRelu<GPUDevice, T>; \
Drelu_op_functor.h96 struct LeakyRelu { struct
Drelu_op.h145 functor::LeakyRelu<Device, T> functor; in Operate()
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad_test.cc167 auto y = ops::internal::LeakyRelu(scope_, x); in TEST_F()
/external/tensorflow/tensorflow/lite/kernels/
Dactivations_test.cc1569 TEST(FloatActivationsOpTest, LeakyRelu) { in TEST() argument
Dactivations.cc1120 optimized_ops::LeakyRelu( in LeakyReluEval()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc1381 class LeakyRelu class
1884 ops.push_back(MakeUnique<LeakyRelu>(::tflite::BuiltinOperator_LEAKY_RELU, in BuildOperatorList()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf.mlir23 func @LeakyRelu(%arg0: tensor<1xf32>) -> tensor<1xf32> {
24 %2 = "tf.LeakyRelu"(%arg0) {alpha = 0.1 : f32} : (tensor<1xf32>) -> tensor<1xf32>
27 // CHECK-LABEL: LeakyRelu
Doptimize.mlir751 // Should not fuse to LeakyRelu, since alpha > 1.
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc440 HANDLE_OPERATORTYPENAME_CASE(LeakyRelu) in OperatorTypeName()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc2637 auto act = ops::internal::LeakyRelu( in TEST_F()
2639 ops::internal::LeakyRelu::Alpha(kLeakyReluAlpha)); in TEST_F()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h254 inline void LeakyRelu(const tflite::LeakyReluParams& params, in LeakyRelu() function
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h86 using reference_ops::LeakyRelu;
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt1952 name: "LeakyRelu"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt1952 name: "LeakyRelu"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td2646 def TF_LeakyReluOp : TF_Op<"LeakyRelu", [NoSideEffect, SameOperandsAndResultType]> {
/external/tensorflow/tensorflow/go/op/
Dwrappers.go41853 func LeakyRelu(scope *Scope, features tf.Output, optional ...LeakyReluAttr) (activations tf.Output)… func

12