Home
last modified time | relevance | path

Searched defs:Relu (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drelu_op_gpu.cu.cc217 struct Relu<Device, qint8> { struct
222 void operator()(const Device& d, typename TTypes<qint8>::ConstTensor input, in operator ()()
262 template struct functor::Relu<GPUDevice, qint8>; member in tensorflow::functor
Drelu_op_functor.h28 struct Relu { struct
45 // gradients: gradients backpropagated to the Relu op. argument
Dfused_eigen_output_kernels.h84 struct Relu { struct
Dfused_batch_norm_ex_op_test.cc567 SideInputAndActivation Relu() { return {false, "Relu"}; } in Relu() function
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Drelu_test.cc97 TEST(Relu, MultiThreading) { in TEST() argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Drelu_op.cc27 XlaOp Relu(XlaOp x) { return Max(ScalarLike(x, 0), x); } in Relu() function
/external/tensorflow/tensorflow/c/experimental/ops/
Dnn_ops.cc87 Status Relu(AbstractContext* ctx, AbstractTensorHandle* const features, in Relu() function
/external/tensorflow/tensorflow/core/transforms/remapper/
Dpass.cc103 enum class OpKind { Relu, Relu6, Elu, LeakyRelu, Tanh, Sigmoid }; enumerator
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dgpu_unary_ops_test.cc896 GENERATE_DEFAULT_TEST(Relu, DT_FLOAT, DT_FLOAT, baseline_relu, in GENERATE_DEFAULT_TEST() argument
/external/tensorflow/tensorflow/core/graph/
Dtestlib.cc442 Node* Relu(Graph* g, Node* in) { in Relu() function
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h95 inline void Relu(const RuntimeShape& input_shape, const T* input_data, in Relu() function
Dlegacy_reference_ops.h1231 inline void Relu(const float* input_data, const Dims<4>& input_dims, in Relu() function
/external/tensorflow/tensorflow/lite/kernels/
Dactivations_test.cc324 TEST(FloatActivationsOpTest, Relu) { in TEST() argument
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2261 TEST(NNAPIDelegate, Relu) { in TEST() argument
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3803 TEST_F(OpTest, Relu) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3486 inline void Relu(const float* input_data, const Dims<4>& input_dims, in Relu() function
Doptimized_ops.h1391 inline void Relu(const RuntimeShape& input_shape, const float* input_data, in Relu() function
/external/tensorflow/tensorflow/go/op/
Dwrappers.go36391 func Relu(scope *Scope, features tf.Output) (activations tf.Output) { func