Searched defs:Relu6 (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | relu_op_functor.h | 62 struct Relu6 { struct
80 // gradients: gradients backpropagated to the Relu6 op. argument
|
| D | fused_eigen_output_kernels.h | 93 struct Relu6 { struct
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | relu6_test.cc | 97 TEST(Relu6, MultiThreading) { in TEST() argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | relu_op.cc | 29 XlaOp Relu6(XlaOp x) { in Relu6() function
|
| /external/tensorflow/tensorflow/core/transforms/remapper/ |
| D | pass.cc | 103 enum class OpKind { Relu, Relu6, Elu, LeakyRelu, Tanh, Sigmoid }; enumerator
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.cc | 451 Node* Relu6(Graph* g, Node* in) { in Relu6() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 134 inline void Relu6(const RuntimeShape& input_shape, const float* input_data, in Relu6() function
|
| D | legacy_reference_ops.h | 1243 inline void Relu6(const float* input_data, const Dims<4>& input_dims, in Relu6() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations_test.cc | 366 TEST(FloatActivationsOpTest, Relu6) { in TEST() argument
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2289 TEST(NNAPIDelegate, Relu6) { in TEST() argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3810 TEST_F(OpTest, Relu6) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 36406 func Relu6(scope *Scope, features tf.Output) (activations tf.Output) { func
|