Home
last modified time | relevance | path

Searched defs:Relu6 (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drelu_op_functor.h62 struct Relu6 { struct
80 // gradients: gradients backpropagated to the Relu6 op. argument
Dfused_eigen_output_kernels.h92 struct Relu6 { struct
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Drelu6_test.cc97 TEST(Relu6, MultiThreading) { in TEST() argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Drelu_op.cc28 XlaOp Relu6(XlaOp x) { in Relu6() function
/external/tensorflow/tensorflow/core/graph/
Dtestlib.cc451 Node* Relu6(Graph* g, Node* in) { in Relu6() function
/external/tensorflow/tensorflow/lite/kernels/
Dactivations_test.cc311 TEST(FloatActivationsOpTest, Relu6) { in TEST() argument
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h116 inline void Relu6(const RuntimeShape& input_shape, const float* input_data, in Relu6() function
Dlegacy_reference_ops.h1240 inline void Relu6(const float* input_data, const Dims<4>& input_dims, in Relu6() function
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2212 TEST(NNAPIDelegate, Relu6) { in TEST() argument
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2635 TEST_F(OpTest, Relu6) { in TEST_F() argument
/external/tensorflow/tensorflow/go/op/
Dwrappers.go28409 func Relu6(scope *Scope, features tf.Output) (activations tf.Output) { func