| 187 | } |
| 188 | |
| 189 | TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input, |
| 190 | TFE_TensorHandle* axis) { |
| 191 | TF_Status* status = TF_NewStatus(); |
| 192 | |
| 193 | TFE_Op* op = TFE_NewOp(ctx, "Min", status); |
| 194 | CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); |
| 195 | TFE_OpAddInput(op, input, status); |
| 196 | CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); |
| 197 | TFE_OpAddInput(op, axis, status); |
| 198 | CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); |
| 199 | TFE_OpSetAttrBool(op, "keep_dims", 1); |
| 200 | TFE_OpSetAttrType(op, "Tidx", TF_INT32); |
| 201 | TF_DeleteStatus(status); |
| 202 | TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(input)); |
| 203 | |
| 204 | return op; |
| 205 | } |
| 206 | |
| 207 | bool GetDeviceName(TFE_Context* ctx, string* device_name, |
| 208 | const char* device_type) { |
no test coverage detected