1 | // This file is MACHINE GENERATED! Do not edit. |
2 | |
3 | |
4 | #include "tensorflow/cc/ops/const_op.h" |
5 | #include "tensorflow/cc/ops/array_ops_internal.h" |
6 | |
7 | namespace tensorflow { |
8 | namespace ops { |
9 | namespace internal { |
10 | // NOTE: This namespace has internal TensorFlow details that |
11 | // are not part of TensorFlow's public API. |
12 | |
13 | BroadcastGradientArgs::BroadcastGradientArgs(const ::tensorflow::Scope& scope, |
14 | ::tensorflow::Input s0, |
15 | ::tensorflow::Input s1) { |
16 | if (!scope.ok()) return; |
17 | auto _s0 = ::tensorflow::ops::AsNodeOut(scope, s0); |
18 | if (!scope.ok()) return; |
19 | auto _s1 = ::tensorflow::ops::AsNodeOut(scope, s1); |
20 | if (!scope.ok()) return; |
21 | ::tensorflow::Node* ret; |
22 | const auto unique_name = scope.GetUniqueNameForOp("BroadcastGradientArgs" ); |
23 | auto builder = ::tensorflow::NodeBuilder(unique_name, "BroadcastGradientArgs" ) |
24 | .Input(_s0) |
25 | .Input(_s1) |
26 | ; |
27 | scope.UpdateBuilder(&builder); |
28 | scope.UpdateStatus(builder.Finalize(scope.graph(), &ret)); |
29 | if (!scope.ok()) return; |
30 | scope.UpdateStatus(scope.DoShapeInference(ret)); |
31 | this->operation = Operation(ret); |
32 | ::tensorflow::NameRangeMap _outputs_range; |
33 | ::tensorflow::Status _status_ = ::tensorflow::NameRangesForNode(*ret, ret->op_def(), nullptr, &_outputs_range); |
34 | if (!_status_.ok()) { |
35 | scope.UpdateStatus(_status_); |
36 | return; |
37 | } |
38 | |
39 | this->r0 = Output(ret, _outputs_range["r0" ].first); |
40 | this->r1 = Output(ret, _outputs_range["r1" ].first); |
41 | } |
42 | |
43 | CheckNumericsV2::CheckNumericsV2(const ::tensorflow::Scope& scope, |
44 | ::tensorflow::Input tensor, StringPiece |
45 | message) { |
46 | if (!scope.ok()) return; |
47 | auto _tensor = ::tensorflow::ops::AsNodeOut(scope, tensor); |
48 | if (!scope.ok()) return; |
49 | ::tensorflow::Node* ret; |
50 | const auto unique_name = scope.GetUniqueNameForOp("CheckNumericsV2" ); |
51 | auto builder = ::tensorflow::NodeBuilder(unique_name, "CheckNumericsV2" ) |
52 | .Input(_tensor) |
53 | .Attr("message" , message) |
54 | ; |
55 | scope.UpdateBuilder(&builder); |
56 | scope.UpdateStatus(builder.Finalize(scope.graph(), &ret)); |
57 | if (!scope.ok()) return; |
58 | scope.UpdateStatus(scope.DoShapeInference(ret)); |
59 | this->operation = Operation(ret); |
60 | this->output = Output(ret, 0); |
61 | } |
62 | |
63 | LowerBound::LowerBound(const ::tensorflow::Scope& scope, ::tensorflow::Input |
64 | sorted_inputs, ::tensorflow::Input values, const |
65 | LowerBound::Attrs& attrs) { |
66 | if (!scope.ok()) return; |
67 | auto _sorted_inputs = ::tensorflow::ops::AsNodeOut(scope, sorted_inputs); |
68 | if (!scope.ok()) return; |
69 | auto _values = ::tensorflow::ops::AsNodeOut(scope, values); |
70 | if (!scope.ok()) return; |
71 | ::tensorflow::Node* ret; |
72 | const auto unique_name = scope.GetUniqueNameForOp("LowerBound" ); |
73 | auto builder = ::tensorflow::NodeBuilder(unique_name, "LowerBound" ) |
74 | .Input(_sorted_inputs) |
75 | .Input(_values) |
76 | .Attr("out_type" , attrs.out_type_) |
77 | ; |
78 | scope.UpdateBuilder(&builder); |
79 | scope.UpdateStatus(builder.Finalize(scope.graph(), &ret)); |
80 | if (!scope.ok()) return; |
81 | scope.UpdateStatus(scope.DoShapeInference(ret)); |
82 | this->operation = Operation(ret); |
83 | this->output = Output(ret, 0); |
84 | } |
85 | |
86 | LowerBound::LowerBound(const ::tensorflow::Scope& scope, ::tensorflow::Input |
87 | sorted_inputs, ::tensorflow::Input values) |
88 | : LowerBound(scope, sorted_inputs, values, LowerBound::Attrs()) {} |
89 | |
90 | MirrorPadGrad::MirrorPadGrad(const ::tensorflow::Scope& scope, |
91 | ::tensorflow::Input input, ::tensorflow::Input |
92 | paddings, StringPiece mode) { |
93 | if (!scope.ok()) return; |
94 | auto _input = ::tensorflow::ops::AsNodeOut(scope, input); |
95 | if (!scope.ok()) return; |
96 | auto _paddings = ::tensorflow::ops::AsNodeOut(scope, paddings); |
97 | if (!scope.ok()) return; |
98 | ::tensorflow::Node* ret; |
99 | const auto unique_name = scope.GetUniqueNameForOp("MirrorPadGrad" ); |
100 | auto builder = ::tensorflow::NodeBuilder(unique_name, "MirrorPadGrad" ) |
101 | .Input(_input) |
102 | .Input(_paddings) |
103 | .Attr("mode" , mode) |
104 | ; |
105 | scope.UpdateBuilder(&builder); |
106 | scope.UpdateStatus(builder.Finalize(scope.graph(), &ret)); |
107 | if (!scope.ok()) return; |
108 | scope.UpdateStatus(scope.DoShapeInference(ret)); |
109 | this->operation = Operation(ret); |
110 | this->output = Output(ret, 0); |
111 | } |
112 | |
113 | RefIdentity::RefIdentity(const ::tensorflow::Scope& scope, ::tensorflow::Input |
114 | input) { |
115 | if (!scope.ok()) return; |
116 | auto _input = ::tensorflow::ops::AsNodeOut(scope, input); |
117 | if (!scope.ok()) return; |
118 | ::tensorflow::Node* ret; |
119 | const auto unique_name = scope.GetUniqueNameForOp("RefIdentity" ); |
120 | auto builder = ::tensorflow::NodeBuilder(unique_name, "RefIdentity" ) |
121 | .Input(_input) |
122 | ; |
123 | scope.UpdateBuilder(&builder); |
124 | scope.UpdateStatus(builder.Finalize(scope.graph(), &ret)); |
125 | if (!scope.ok()) return; |
126 | scope.UpdateStatus(scope.DoShapeInference(ret)); |
127 | this->operation = Operation(ret); |
128 | this->output = Output(ret, 0); |
129 | } |
130 | |
131 | UpperBound::UpperBound(const ::tensorflow::Scope& scope, ::tensorflow::Input |
132 | sorted_inputs, ::tensorflow::Input values, const |
133 | UpperBound::Attrs& attrs) { |
134 | if (!scope.ok()) return; |
135 | auto _sorted_inputs = ::tensorflow::ops::AsNodeOut(scope, sorted_inputs); |
136 | if (!scope.ok()) return; |
137 | auto _values = ::tensorflow::ops::AsNodeOut(scope, values); |
138 | if (!scope.ok()) return; |
139 | ::tensorflow::Node* ret; |
140 | const auto unique_name = scope.GetUniqueNameForOp("UpperBound" ); |
141 | auto builder = ::tensorflow::NodeBuilder(unique_name, "UpperBound" ) |
142 | .Input(_sorted_inputs) |
143 | .Input(_values) |
144 | .Attr("out_type" , attrs.out_type_) |
145 | ; |
146 | scope.UpdateBuilder(&builder); |
147 | scope.UpdateStatus(builder.Finalize(scope.graph(), &ret)); |
148 | if (!scope.ok()) return; |
149 | scope.UpdateStatus(scope.DoShapeInference(ret)); |
150 | this->operation = Operation(ret); |
151 | this->output = Output(ret, 0); |
152 | } |
153 | |
154 | UpperBound::UpperBound(const ::tensorflow::Scope& scope, ::tensorflow::Input |
155 | sorted_inputs, ::tensorflow::Input values) |
156 | : UpperBound(scope, sorted_inputs, values, UpperBound::Attrs()) {} |
157 | |
158 | } // namespace internal |
159 | } // namespace ops |
160 | } // namespace tensorflow |
161 | |