Home
last modified time | relevance | path

Searched defs:Relu (Results 1 – 16 of 16) 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 ()()
254 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.h83 struct Relu { struct
Dfused_batch_norm_ex_op_test.cc546 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/c/experimental/ops/
Dnn_ops.cc59 Status Relu(AbstractContext* ctx, in Relu() function
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Drelu_op.cc26 XlaOp Relu(XlaOp x) { return Max(ScalarLike(x, 0), x); } in Relu() function
/external/tensorflow/tensorflow/core/graph/
Dtestlib.cc442 Node* Relu(Graph* g, Node* in) { in Relu() function
/external/tensorflow/tensorflow/lite/kernels/
Dactivations_test.cc283 TEST(FloatActivationsOpTest, Relu) { in TEST() argument
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h91 inline void Relu(const RuntimeShape& input_shape, const T* input_data, in Relu() function
Dlegacy_reference_ops.h1228 inline void Relu(const float* input_data, const Dims<4>& input_dims, in Relu() function
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2184 TEST(NNAPIDelegate, Relu) { in TEST() argument
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2628 TEST_F(OpTest, Relu) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3485 inline void Relu(const float* input_data, const Dims<4>& input_dims, in Relu() function
Doptimized_ops.h1722 inline void Relu(const RuntimeShape& input_shape, const float* input_data, in Relu() function
/external/tensorflow/tensorflow/go/op/
Dwrappers.go32576 func Relu(scope *Scope, features tf.Output) (activations tf.Output) { func