Home
last modified time | relevance | path

Searched defs:relu6 (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dquantize_training_test.cc144 Node* relu6 = test::graph::Relu6(g, b); in TEST_F() local
374 Node* relu6 = test::graph::Relu6(g, c); in TEST_F() local
468 Node* relu6 = test::graph::Relu6(g, c); in TEST_F() local
Dmkl_layout_pass.cc954 string relu6; member
/external/tensorflow/tensorflow/python/keras/
Dactivations.py516 relu6 = nn.relu6 variable
/external/pytorch/torch/csrc/api/include/torch/nn/functional/
Dactivation.h437 inline Tensor relu6(Tensor input, bool inplace) { in relu6() function
/external/pytorch/torch/_refs/nn/functional/
D__init__.py1155 def relu6(a: TensorLikeType, inplace: bool = False) -> TensorLikeType: function
/external/pytorch/aten/src/ATen/native/
DActivation.cpp525 Tensor relu6(const Tensor & self) { in relu6() function
/external/pytorch/torch/onnx/
Dsymbolic_opset11.py178 def relu6(g: jit_utils.GraphContext, input): function
Dsymbolic_opset9.py1225 def relu6(g: jit_utils.GraphContext, input): function
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py3627 def relu6(features, name=None): function
/external/pytorch/torch/nn/
Dfunctional.py1780 def relu6(input: Tensor, inplace: bool = False) -> Tensor: # noqa: D400,D402 function