1 | // This file is MACHINE GENERATED! Do not edit. |
2 | |
3 | #ifndef TENSORFLOW_CC_OPS_STATE_OPS_INTERNAL_H_ |
4 | #define TENSORFLOW_CC_OPS_STATE_OPS_INTERNAL_H_ |
5 | |
6 | // This file is MACHINE GENERATED! Do not edit. |
7 | |
8 | #include "tensorflow/cc/framework/ops.h" |
9 | #include "tensorflow/cc/framework/scope.h" |
10 | #include "tensorflow/core/framework/tensor.h" |
11 | #include "tensorflow/core/framework/tensor_shape.h" |
12 | #include "tensorflow/core/framework/types.h" |
13 | #include "tensorflow/core/lib/gtl/array_slice.h" |
14 | |
15 | namespace tensorflow { |
16 | namespace ops { |
17 | namespace internal { |
18 | // NOTE: This namespace has internal TensorFlow details that |
19 | // are not part of TensorFlow's public API. |
20 | |
21 | /// @defgroup state_ops_internal State Ops Internal |
22 | /// @{ |
23 | |
24 | /// Computes element-wise maximum. |
25 | /// |
26 | /// Args: |
27 | /// * scope: A Scope object |
28 | /// * ref: A mutable Tensor. Should be from a Variable node. |
29 | /// * indices: A Tensor. Must be one of the following types: int32, int64. |
30 | /// A tensor of indices into ref. |
31 | /// * updates: A Tensor. Must have the same type as ref. A tensor of updated values |
32 | /// to add to ref. |
33 | /// |
34 | /// Optional attributes (see `Attrs`): |
35 | /// * use_locking: An optional bool. Defaults to True. If True, the assignment will |
36 | /// be protected by a lock; otherwise the behavior is undefined, |
37 | /// but may exhibit less contention. |
38 | /// |
39 | /// Returns: |
40 | /// * `Output`: Same as ref. Returned as a convenience for operations that want |
41 | /// to use the updated values after the update is done. |
42 | class ScatterNdMax { |
43 | public: |
44 | /// Optional attribute setters for ScatterNdMax |
45 | struct Attrs { |
46 | /// An optional bool. Defaults to True. If True, the assignment will |
47 | /// be protected by a lock; otherwise the behavior is undefined, |
48 | /// but may exhibit less contention. |
49 | /// |
50 | /// Defaults to false |
51 | TF_MUST_USE_RESULT Attrs UseLocking(bool x) { |
52 | Attrs ret = *this; |
53 | ret.use_locking_ = x; |
54 | return ret; |
55 | } |
56 | |
57 | bool use_locking_ = false; |
58 | }; |
59 | ScatterNdMax(const ::tensorflow::Scope& scope, ::tensorflow::Input ref, |
60 | ::tensorflow::Input indices, ::tensorflow::Input updates); |
61 | ScatterNdMax(const ::tensorflow::Scope& scope, ::tensorflow::Input ref, |
62 | ::tensorflow::Input indices, ::tensorflow::Input updates, const |
63 | ScatterNdMax::Attrs& attrs); |
64 | operator ::tensorflow::Output() const { return output_ref; } |
65 | operator ::tensorflow::Input() const { return output_ref; } |
66 | ::tensorflow::Node* node() const { return output_ref.node(); } |
67 | |
68 | static Attrs UseLocking(bool x) { |
69 | return Attrs().UseLocking(x); |
70 | } |
71 | |
72 | Operation operation; |
73 | ::tensorflow::Output output_ref; |
74 | }; |
75 | |
76 | /// Computes element-wise minimum. |
77 | /// |
78 | /// Args: |
79 | /// * scope: A Scope object |
80 | /// * ref: A mutable Tensor. Should be from a Variable node. |
81 | /// * indices: A Tensor. Must be one of the following types: int32, int64. |
82 | /// A tensor of indices into ref. |
83 | /// * updates: A Tensor. Must have the same type as ref. A tensor of updated values |
84 | /// to add to ref. |
85 | /// |
86 | /// Optional attributes (see `Attrs`): |
87 | /// * use_locking: An optional bool. Defaults to True. If True, the assignment will |
88 | /// be protected by a lock; otherwise the behavior is undefined, |
89 | /// but may exhibit less contention. |
90 | /// |
91 | /// Returns: |
92 | /// * `Output`: Same as ref. Returned as a convenience for operations that want |
93 | /// to use the updated values after the update is done. |
94 | class ScatterNdMin { |
95 | public: |
96 | /// Optional attribute setters for ScatterNdMin |
97 | struct Attrs { |
98 | /// An optional bool. Defaults to True. If True, the assignment will |
99 | /// be protected by a lock; otherwise the behavior is undefined, |
100 | /// but may exhibit less contention. |
101 | /// |
102 | /// Defaults to false |
103 | TF_MUST_USE_RESULT Attrs UseLocking(bool x) { |
104 | Attrs ret = *this; |
105 | ret.use_locking_ = x; |
106 | return ret; |
107 | } |
108 | |
109 | bool use_locking_ = false; |
110 | }; |
111 | ScatterNdMin(const ::tensorflow::Scope& scope, ::tensorflow::Input ref, |
112 | ::tensorflow::Input indices, ::tensorflow::Input updates); |
113 | ScatterNdMin(const ::tensorflow::Scope& scope, ::tensorflow::Input ref, |
114 | ::tensorflow::Input indices, ::tensorflow::Input updates, const |
115 | ScatterNdMin::Attrs& attrs); |
116 | operator ::tensorflow::Output() const { return output_ref; } |
117 | operator ::tensorflow::Input() const { return output_ref; } |
118 | ::tensorflow::Node* node() const { return output_ref.node(); } |
119 | |
120 | static Attrs UseLocking(bool x) { |
121 | return Attrs().UseLocking(x); |
122 | } |
123 | |
124 | Operation operation; |
125 | ::tensorflow::Output output_ref; |
126 | }; |
127 | |
128 | } // namespace internal |
129 | } // namespace ops |
130 | } // namespace tensorflow |
131 | |
132 | #endif // TENSORFLOW_CC_OPS_STATE_OPS_INTERNAL_H_ |
133 | |