Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/graph/
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 …]
Dmkl_layout_pass.cc317 csinfo_.relu6 = "Relu6"; in MklLayoutRewritePass()
499 rinfo_.push_back({csinfo_.relu6, in MklLayoutRewritePass()
500 mkl_op_registry::GetMklOpName(csinfo_.relu6), in MklLayoutRewritePass()
741 string relu6; member
/external/tensorflow/tensorflow/contrib/quantize/python/
Dquantize_test.py58 relu = nn_ops.relu6(inputs)
81 node = nn_ops.relu6(node, name='test/relu6')
121 node = nn_ops.relu6(node, name='test/relu6')
163 node = nn_ops.relu6(node, name='test/relu6')
213 activation_fn=nn_ops.relu6,
263 activation_fn=nn_ops.relu6,
273 activation_fn=nn_ops.relu6,
301 activation_fn=nn_ops.relu6,
316 _ = nn_ops.relu6(bypass_tensor, name='test/output')
438 node = nn_ops.relu6(node, name='test/relu6')
[all …]
Dquantize_parameterized_test.py46 (nn_ops.relu6, 'Relu6', False, None),
49 (nn_ops.relu6, 'Relu6', False, 5000),
52 (nn_ops.relu6, 'Relu6', True, None),
55 (nn_ops.relu6, 'Relu6', True, 5000),
366 (nn_ops.relu6, 'Relu6', False, None, False),
369 (nn_ops.relu6, 'Relu6', False, 5000, False),
372 (nn_ops.relu6, 'Relu6', True, None, False),
375 (nn_ops.relu6, 'Relu6', True, 5000, False),
378 (nn_ops.relu6, 'Relu6', False, None, True),
381 (nn_ops.relu6, 'Relu6', False, 5000, True),
[all …]
Dinput_to_ops_test.py45 output_tensor = nn_ops.relu6(input_tensor)
56 output_tensor_1 = nn_ops.relu6(input_tensor)
Dfold_batch_norms_test.py52 (nn_ops.relu6, 'Relu6', False, False, False, 100, False),
54 (nn_ops.relu6, 'Relu6', True, False, False, 100, False),
56 (nn_ops.relu6, 'Relu6', False, True, False, 100, False),
58 (nn_ops.relu6, 'Relu6', True, True, False, 100, False),
61 (nn_ops.relu6, 'Relu6', False, True, True, None, False),
63 (nn_ops.relu6, 'Relu6', True, True, True, None, False),
65 (nn_ops.relu6, 'Relu6', False, True, True, None, True),
67 (nn_ops.relu6, 'Relu6', True, True, True, None, True),
Dquantize_graph_test.py370 output = nn_ops.relu6(output)
454 output = nn_ops.relu6(output)
509 output = nn_ops.relu6(output)
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
Dnn.py26 relu6 = core.define_unary_op('relu6', nn.relu6) variable
Dnn_test.py38 ('relu6', nn_ops.relu6, nn.relu6),
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py98 x = nn.relu6(x)
101 x = nn.relu6(x)
107 x = nn.relu6(x)
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_test.py422 def relu6(x): function
425 outputs = keras.layers.Activation(relu6)(inputs)
428 model, saved_model_dir, custom_objects={'relu6': relu6})
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dsummaries_test.py69 op = nn_ops.relu6(var, name='SummaryTest')
Dlayers.py3324 relu6 = functools.partial(fully_connected, activation_fn=nn.relu6) variable
/external/tensorflow/tensorflow/python/kernel_tests/
Drelu_op_test.py237 tf_relu6 = nn_ops.relu6(np_features)
265 *gradient_checker_v2.compute_gradient(nn_ops.relu6, [x]))
276 *gradient_checker_v2.compute_gradient(nn_ops.relu6, [x]))
/external/tensorflow/tensorflow/python/ops/
Dnn_grad_test.py42 r = nn_ops.relu6(inputs)
Dnn_ops.py2719 def relu6(features, name=None): function
2735 return gen_nn_ops.relu6(features, name=name)
/external/tensorflow/tensorflow/core/kernels/
Dunary_ops_composition.cc185 auto relu6 = functor::Relu6<Eigen::DefaultDevice, T>(); \
186 relu6(Eigen::DefaultDevice(), in, *out); \
/external/tensorflow/tensorflow/python/tools/
Dfreeze_graph_test.py293 node = nn.relu6(node, name="test/relu6")
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dmath_test.py86 nn.relu6,
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_vectorization_test.py87 ("Relu6", nn.relu6),
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_compatibility.md125 * [tf.nn.relu6](https://www.tensorflow.org/api_docs/python/tf/nn/relu6)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt248 name: "relu6"
/external/tensorflow/tensorflow/lite/
Dbuild_def.bzl300 "relu6",
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt320 name: "relu6"
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_test.py359 nn_ops.relu6,

12