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
15namespace tensorflow {
16namespace ops {
17namespace 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.
42class 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.
94class 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