Home
last modified time | relevance | path

Searched refs:relu6 (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/
Dquantize_training_test.cc144 Node* relu6 = test::graph::Relu6(g, b); in TEST_F() local
145 Node* m1 = test::graph::Matmul(g, relu, relu6, false, false); in TEST_F()
165 FindNode(g, strings::StrCat(relu6->name(), "/QuantizeAndDequantizeV2"), in TEST_F()
374 Node* relu6 = test::graph::Relu6(g, c); in TEST_F() local
375 Node* m1 = test::graph::Matmul(g, relu, relu6, false, false); in TEST_F()
403 string min_const_name = strings::StrCat(relu6->name(), "/InputMin"); in TEST_F()
404 string max_const_name = strings::StrCat(relu6->name(), "/InputMax"); in TEST_F()
468 Node* relu6 = test::graph::Relu6(g, c); in TEST_F() local
469 Node* m1 = test::graph::Matmul(g, relu, relu6, false, false); in TEST_F()
497 string min_const_name = strings::StrCat(relu6->name(), "/InputMin"); in TEST_F()
[all …]
/external/tensorflow/tensorflow/python/keras/
Dactivations.py516 relu6 = nn.relu6 variable
/external/tensorflow/tensorflow/compiler/mlir/tfr/resources/
Ddecomposition_lib.mlir41 %relu6 = tfr.constant "relu6" -> !tfr.attr
48 %is_relu6 = tfr.equal %act, %relu6 -> i1
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/
Dquantize_model_test.py227 'activation_fn': [None, nn_ops.relu, nn_ops.relu6],
418 ('relu6', nn_ops.relu6, False, False, quant_opts_pb2.TF, False),
427 ('with_bias_and_relu6', nn_ops.relu6, True, False, quant_opts_pb2.TF,
430 ('with_bias_and_relu6_to_xla', nn_ops.relu6, True, False,
434 ('with_bias_and_relu6_to_xla_dynamic', nn_ops.relu6, True, False,
524 ('relu6', nn_ops.relu6, False, False, quant_opts_pb2.TF, False),
533 ('with_bias_and_relu6', nn_ops.relu6, True, False, quant_opts_pb2.TF,
536 ('with_bias_and_relu6_to_xla', nn_ops.relu6, True, False,
540 ('with_bias_and_relu6_to_xla_dynamic', nn_ops.relu6, True, False,
631 ('relu6', nn_ops.relu6, False),
[all …]
Dquantize_model_test_base.py339 out = nn_ops.relu6(out)
/external/tensorflow/tensorflow/lite/testing/op_tests/
Drelu6.py38 out = tf.nn.relu6(input_tensor)
Dhardswish.py62 out = inp * tf.nn.relu6(inp + np.float32(3)) * np.float32(1. / 6.)
Dconv_activation.py133 return make_conv_activation_tests(tf.nn.relu6)(options)
Dconv_bias_activation.py145 return make_conv_bias_activation_tests(tf.nn.relu6)(options)
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py125 x = nn.relu6(x)
128 x = nn.relu6(x)
134 x = nn.relu6(x)
/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/
Dconv_bias_activation.py149 return make_conv_bias_activation_tests(tf.nn.relu6)(options)
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Drelu_op_test.py220 tf_relu6 = nn_ops.relu6(np_features)
253 *gradient_checker_v2.compute_gradient(nn_ops.relu6, [x]))
263 *gradient_checker_v2.compute_gradient(nn_ops.relu6, [x]))
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_test_ops.py82 nn_ops.relu6,
/external/tensorflow/tensorflow/core/kernels/
Dunary_ops_composition.cc186 auto relu6 = functor::Relu6<Eigen::DefaultDevice, T>(); \
187 relu6(Eigen::DefaultDevice(), in, *out); \
/external/tensorflow/tensorflow/compiler/mlir/tfr/tests/
Ddecompose.mlir177 …%relu6:2 = "tfr.quant_act_range"(%relu6_attr, %scale, %zp) : (!tfr.attr, f32, i64) -> (!tfr.tensor…
181 %relu6#0, %relu6#1, %reluN1_1#0, %reluN1_1#1) : (
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_compatibility.md77 * `tf.nn.relu6`
/external/tensorflow/tensorflow/python/ops/
Dnn_grad_test.py64 r = nn_ops.relu6(inputs)
/external/tensorflow/tensorflow/python/grappler/
Dremapper_test.py204 act_fns += [nn.elu, nn.relu6, nn.leaky_relu]
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dlift_quantizable_spots_as_functions.td214 defm : LiftCompositeOpsWithActivation<TF_Relu6Op, "relu6">;
Dquantized_function_library.mlir149 %relu6 = "tf.Relu6"(%dequantize) : (tensor<*xf32>) -> tensor<*xf32>
150 func.return %relu6 : tensor<*xf32>
/external/tensorflow/tensorflow/python/tools/
Dfreeze_graph_test.py299 node = nn.relu6(node, name="test/relu6")
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples_lib.py140 from tensorflow.lite.testing.op_tests.relu6 import make_relu6_tests
DBUILD663 "relu6",
Dbuild_def.bzl128 "relu6",
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt268 name: "relu6"

12