| 129 | CreateTestVariantOp); |
| 130 | |
| 131 | class CreateTestVariant { |
| 132 | public: |
| 133 | explicit CreateTestVariant(const ::tensorflow::Scope& scope, |
| 134 | const Input& value) { |
| 135 | if (!scope.ok()) return; |
| 136 | auto _value = ops::AsNodeOut(scope, value); |
| 137 | if (!scope.ok()) return; |
| 138 | ::tensorflow::Node* ret; |
| 139 | const auto unique_name = scope.GetUniqueNameForOp("CreateTestVariant"); |
| 140 | auto builder = ::tensorflow::NodeBuilder(unique_name, "CreateTestVariant") |
| 141 | .Input(_value); |
| 142 | scope.UpdateBuilder(&builder); |
| 143 | scope.UpdateStatus(builder.Finalize(scope.graph(), &ret)); |
| 144 | if (!scope.ok()) return; |
| 145 | scope.UpdateStatus(scope.DoShapeInference(ret)); |
| 146 | if (!scope.ok()) return; |
| 147 | this->output_ = Output(ret, 0); |
| 148 | } |
| 149 | |
| 150 | // Intentionally not marked as explicit. |
| 151 | // NOLINTNEXTLINE google-explicit-constructor |
| 152 | operator ::tensorflow::Output() const { return output_; } |
| 153 | // Intentionally not marked as explicit. |
| 154 | // NOLINTNEXTLINE google-explicit-constructor |
| 155 | operator ::tensorflow::Input() const { return output_; } |
| 156 | |
| 157 | ::tensorflow::Node* node() const { return output_.node(); } |
| 158 | |
| 159 | ::tensorflow::Output output_; |
| 160 | }; |
| 161 | |
| 162 | } // end namespace |
| 163 | |