| /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/c/experimental/ops/ |
| D | nn_ops.cc | 62 AbstractOperationPtr relu_op(ctx->CreateOperation()); in Relu() local
63 TF_RETURN_IF_ERROR(relu_op->Reset("Relu", /*raw_device_name=*/nullptr)); in Relu()
64 TF_RETURN_IF_ERROR(MaybeSetOpName(relu_op.get(), name)); in Relu()
65 TF_RETURN_IF_ERROR(relu_op->AddInput(inputs[0])); in Relu()
68 TF_RETURN_IF_ERROR(relu_op->Execute(outputs, &num_retvals)); in Relu()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | gpu_fusion.cc | 64 LogicalResult matchAndRewrite(ReluOp relu_op, in matchAndRewrite()
66 Operation *relu_input = relu_op.features().getDefiningOp(); in matchAndRewrite()
108 rewriter.replaceOp(relu_op, op->getResult(0)); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_test.py | 510 relu_op = backend.relu(x)
511 self.assertAllClose(backend.eval(relu_op), [[0, 0], [2, 7]])
514 relu_op = backend.relu(x, alpha=0.5)
516 self.assertTrue('LeakyRelu' in relu_op.name)
517 self.assertAllClose(backend.eval(relu_op), [[-2, 0], [2, 7]])
520 relu_op = backend.relu(x, max_value=5)
521 self.assertAllClose(backend.eval(relu_op), [[0, 0], [2, 5]])
524 relu_op = backend.relu(x, max_value=6)
526 self.assertTrue('Relu6' in relu_op.name) # uses tf.nn.relu6
527 self.assertAllClose(backend.eval(relu_op), [[0, 0], [2, 6]])
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 408 LogicalResult matchAndRewrite(ReluXOp relu_op, in matchAndRewrite()
410 Operation *input = relu_op.getOperand().getDefiningOp(); in matchAndRewrite()
422 FusedLoc::get({fully_connected_op.getLoc(), relu_op.getLoc()}, in matchAndRewrite()
423 relu_op.getContext()), in matchAndRewrite()
424 relu_op.getType(), fully_connected_op.input(), in matchAndRewrite()
427 rewriter.replaceOp(relu_op, fc.output()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 731 tensorflow::NodeDef* relu_op = tensorflow_graph->add_node(); in ConvertReluOperator() local
732 relu_op->set_op("Relu"); in ConvertReluOperator()
733 relu_op->set_name(src_op.outputs[0]); in ConvertReluOperator()
734 *relu_op->add_input() = src_op.inputs[0]; in ConvertReluOperator()
735 (*relu_op->mutable_attr())["T"].set_type( in ConvertReluOperator()
780 tensorflow::NodeDef* relu_op = tensorflow_graph->add_node(); in ConvertRelu6Operator() local
781 relu_op->set_op("Relu6"); in ConvertRelu6Operator()
782 relu_op->set_name(src_op.outputs[0]); in ConvertRelu6Operator()
783 *relu_op->add_input() = src_op.inputs[0]; in ConvertRelu6Operator()
784 (*relu_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertRelu6Operator()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 85 "relu_op.cc",
86 "relu_op.h",
|
| /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/mlir/tosa/transforms/ |
| D | legalize_common.cc | 2233 auto relu_op = rewriter.create<tosa::ReluNOp>( in convertFusedActivation() local
2238 op2_relu_op1 = relu_op.getResult(); in convertFusedActivation()
2244 auto relu_op = rewriter.create<tosa::ReluNOp>( in convertFusedActivation() local
2249 op2_relu_op1 = relu_op.getResult(); in convertFusedActivation()
2257 auto relu_op = rewriter.create<tosa::ClampOp>( in convertFusedActivation() local
2263 op2_relu_op1 = relu_op.getResult(); in convertFusedActivation()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | BUILD | 430 "//tensorflow/core/kernels:relu_op",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 1709 ":relu_op",
1866 ":relu_op",
3424 ":relu_op",
3892 ":relu_op",
3974 name = "relu_op",
3975 prefix = "relu_op",
5933 "relu_op.h",
6186 "relu_op.cc",
|