Searched refs:relu_op (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/python/ |
| D | transform_graph_test.py | 64 relu_op = input_graph_def.node.add()
65 relu_op.op = "Relu"
66 relu_op.attr["T"].CopyFrom(attr_value_pb2.AttrValue(
68 relu_op.name = "relu_op"
69 relu_op.input.extend(["add_op"])
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_test.py | 378 relu_op = keras.backend.relu(x)
379 self.assertAllClose(keras.backend.eval(relu_op), [[0, 0], [2, 7]])
382 relu_op = keras.backend.relu(x, alpha=0.5)
384 self.assertTrue('LeakyRelu' in relu_op.name)
385 self.assertAllClose(keras.backend.eval(relu_op), [[-2, 0], [2, 7]])
388 relu_op = keras.backend.relu(x, max_value=5)
389 self.assertAllClose(keras.backend.eval(relu_op), [[0, 0], [2, 5]])
392 relu_op = keras.backend.relu(x, max_value=6)
394 self.assertTrue('Relu6' in relu_op.name) # uses tf.nn.relu6
395 self.assertAllClose(keras.backend.eval(relu_op), [[0, 0], [2, 6]])
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 726 tensorflow::NodeDef* relu_op = tensorflow_graph->add_node(); in ConvertReluOperator() local
727 relu_op->set_op("Relu"); in ConvertReluOperator()
728 relu_op->set_name(src_op.outputs[0]); in ConvertReluOperator()
729 *relu_op->add_input() = src_op.inputs[0]; in ConvertReluOperator()
730 (*relu_op->mutable_attr())["T"].set_type( in ConvertReluOperator()
775 tensorflow::NodeDef* relu_op = tensorflow_graph->add_node(); in ConvertRelu6Operator() local
776 relu_op->set_op("Relu6"); in ConvertRelu6Operator()
777 relu_op->set_name(src_op.outputs[0]); in ConvertRelu6Operator()
778 *relu_op->add_input() = src_op.inputs[0]; in ConvertRelu6Operator()
779 (*relu_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertRelu6Operator()
[all …]
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 404 Output relu_op = Relu(root.WithOpName("relu_op"), constant_op); in TestQuantizeRelu() local
767 Output relu_op = Relu(root.WithOpName("relu_op"), dequantize_op); in TestRemoveRedundantQuantizationWithMultipleOutputs() local
791 Output relu_op = Relu(root.WithOpName("relu_op"), placeholder_op); in TestQuantizePlaceholders() local
1093 Output relu_op = Relu(root.WithOpName("relu_op"), bias_add_op); in TestHoistFakeQuants() local
1095 Output max_pool_op = MaxPool(root.WithOpName("max_pool_op"), relu_op, in TestHoistFakeQuants()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 73 "relu_op.cc",
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 202 tensorflow/core/kernels/relu_op.cc
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3428 ":relu_op",
3870 ":relu_op",
3943 name = "relu_op",
3944 prefix = "relu_op",
5621 "relu_op.h",
5795 "relu_op.cc",
|