| /external/tensorflow/tensorflow/core/kernels/ |
| relu_op_gpu.cu.cc | 151 template struct functor::Relu6<GPUDevice, T>; \
|
| unary_ops_composition.cc | 170 // Register compute function for the Relu/Relu6/Elu/Selu.
185 auto relu6 = functor::Relu6<Eigen::DefaultDevice, T>(); \
186 relu6(Eigen::DefaultDevice(), in, *out); \
253 REGISTER_COMPUTE_FN(Relu6);
318 REGISTER_COMPUTE_FN(Relu6);
382 REGISTER_COMPUTE_FN(Relu6);
|
| relu_op.cc | 44 Name("Relu6").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
95 void Relu6<GPUDevice, T>::operator()( \
98 extern template struct Relu6<GPUDevice, T>; \
164 Name("Relu6").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
224 Name("Relu6").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
|
| relu_op_functor.h | 61 struct Relu6 {
62 // Computes Relu6 activation.
78 // gradients: gradients backpropagated to the Relu6 op.
79 // features: inputs that where passed to the Relu6 op, or its outputs.
80 // backprops: gradients to backpropagate to the Relu6 inputs.
87 // the relu6.
|
| relu_op.h | 98 functor::Relu6<Device, T> functor;
|
| /external/tensorflow/tensorflow/core/graph/ |
| quantize_training_test.cc | 133 Relu Relu6
144 Node* relu6 = test::graph::Relu6(g, b); local
145 Node* m1 = test::graph::Matmul(g, relu, relu6, false, false);
153 Relu Relu6
162 // Quantize_and_dequantize node for relu6 should have range_given==true.
165 FindNode(g, strings::StrCat(relu6->name(), "/QuantizeAndDequantizeV2"),
362 Relu Relu6
374 Node* relu6 = test::graph::Relu6(g, c) local
468 Node* relu6 = test::graph::Relu6(g, c); local
[all...] |
| testlib.h | 201 // Add a Relu6 node in "g".
202 Node* Relu6(Graph* g, Node* in);
|
| testlib.cc | 450 Node* Relu6(Graph* g, Node* in) {
452 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Relu6")
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| nn_grad_test.cc | 44 using ops::Relu6;
155 auto y = Relu6(scope_, x);
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| activations_test.cc | 163 TEST(FloatActivationsOpTest, Relu6) {
[all...] |
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| operator.cc | [all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| quantize_nodes_test.cc | 110 "Relu6", "ResizeBilinear", "AvgPool", "MaxPool", "Mul"};
422 Output relu6_op = Relu6(root.WithOpName("relu6_op"), constant_op);
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| legacy_reference_ops.h | [all...] |
| reference_ops.h | 427 inline void Relu6(const RuntimeShape& input_shape, const float* input_data,
429 gemmlowp::ScopedProfilingLabel label("Relu6 (not fused)");
[all...] |
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| nnapi_delegate_test.cc | [all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| legacy_optimized_ops.h | 67 using reference_ops::Relu6;
[all...] |
| optimized_ops.h | 84 using reference_ops::Relu6;
[all...] |
| /external/tensorflow/tensorflow/lite/toco/ |
| tooling_util.cc | 337 HANDLE_OPERATORTYPENAME_CASE(Relu6)
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| convert_nodes_test.cc | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| optimized_ops.h | 78 using reference_ops::Relu6;
[all...] |
