1 | // This file is MACHINE GENERATED! Do not edit. |
---|---|
2 | |
3 | |
4 | #include "tensorflow/cc/ops/const_op.h" |
5 | #include "tensorflow/cc/ops/manip_ops.h" |
6 | |
7 | namespace tensorflow { |
8 | namespace ops { |
9 | |
10 | Roll::Roll(const ::tensorflow::Scope& scope, ::tensorflow::Input input, |
11 | ::tensorflow::Input shift, ::tensorflow::Input axis) { |
12 | if (!scope.ok()) return; |
13 | auto _input = ::tensorflow::ops::AsNodeOut(scope, input); |
14 | if (!scope.ok()) return; |
15 | auto _shift = ::tensorflow::ops::AsNodeOut(scope, shift); |
16 | if (!scope.ok()) return; |
17 | auto _axis = ::tensorflow::ops::AsNodeOut(scope, axis); |
18 | if (!scope.ok()) return; |
19 | ::tensorflow::Node* ret; |
20 | const auto unique_name = scope.GetUniqueNameForOp("Roll"); |
21 | auto builder = ::tensorflow::NodeBuilder(unique_name, "Roll") |
22 | .Input(_input) |
23 | .Input(_shift) |
24 | .Input(_axis) |
25 | ; |
26 | scope.UpdateBuilder(&builder); |
27 | scope.UpdateStatus(builder.Finalize(scope.graph(), &ret)); |
28 | if (!scope.ok()) return; |
29 | scope.UpdateStatus(scope.DoShapeInference(ret)); |
30 | this->operation = Operation(ret); |
31 | this->output = Output(ret, 0); |
32 | } |
33 | |
34 | /// @} |
35 | |
36 | } // namespace ops |
37 | } // namespace tensorflow |
38 |