| 724 | } |
| 725 | |
| 726 | int TF_OpIsStateful(const char* op_type, TF_Status* status) { |
| 727 | const tensorflow::OpRegistrationData* op_reg_data; |
| 728 | status->status = |
| 729 | tensorflow::OpRegistry::Global()->LookUp(op_type, &op_reg_data); |
| 730 | if (!status->status.ok()) { |
| 731 | return 0; |
| 732 | } |
| 733 | return op_reg_data->op_def.is_stateful(); |
| 734 | } |
| 735 | |
| 736 | void TF_InitMain(const char* usage, int* argc, char*** argv) { |
| 737 | tensorflow::port::InitMain(usage, argc, argv); |