Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Drelu_op_gpu.cu.cc143 struct Relu<Device, qint8> { struct
148 void operator()(const Device& d, typename TTypes<qint8>::ConstTensor input, in operator ()()
182 template struct functor::Relu<GPUDevice, qint8>; member in tensorflow::functor
Drelu_op_functor.h28 struct Relu { struct
44 // gradients: gradients backpropagated to the Relu op. argument
Dfused_eigen_output_kernels.h80 struct Relu { struct
Dfused_batch_norm_ex_op_test.cc546 SideInputAndActivation Relu() { return {false, "Relu"}; } 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.cc221 TEST(FloatActivationsOpTest, Relu) { in TEST() argument
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h182 inline void Relu(const RuntimeShape& input_shape, const T* input_data, in Relu() function
Dlegacy_reference_ops.h1258 inline void Relu(const float* input_data, const Dims<4>& input_dims, in Relu() function
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc1970 TEST(NNAPIDelegate, Relu) { in TEST() argument
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2627 TEST_F(OpTest, Relu) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3472 inline void Relu(const float* input_data, const Dims<4>& input_dims, in Relu() function
Doptimized_ops.h1589 inline void Relu(const RuntimeShape& input_shape, const float* input_data, in Relu() function
/external/tensorflow/tensorflow/go/op/
Dwrappers.go30396 func Relu(scope *Scope, features tf.Output) (activations tf.Output) { func