| 59 | } |
| 60 | |
| 61 | Status MoveSsdFiles(Env* env, |
| 62 | const gtl::ArraySlice<tstring>& input_prefixes, |
| 63 | const tstring& merged_prefix) { |
| 64 | for (auto input_prefix : input_prefixes) { |
| 65 | const tstring& input_ssd_record_pattern = |
| 66 | input_prefix + "*-ssd_record*"; |
| 67 | TF_RETURN_IF_ERROR(MoveMatchingFiles( |
| 68 | env, input_ssd_record_pattern, merged_prefix, input_prefix.size())); |
| 69 | |
| 70 | const tstring& input_emb_files_pattern = |
| 71 | input_prefix + "*-emb_files"; |
| 72 | TF_RETURN_IF_ERROR(MoveMatchingFiles( |
| 73 | env, input_emb_files_pattern, merged_prefix, input_prefix.size())); |
| 74 | } |
| 75 | return Status::OK(); |
| 76 | } |
| 77 | |
| 78 | #define REGISTER_KV_VAR_HANDLE(dev, ktype, vtype) \ |
| 79 | REGISTER_KERNEL_BUILDER(Name("KvVarHandleOp") \ |
no test coverage detected