| 42 | |
| 43 | |
| 44 | TEST(KernelsTest, KvImportTest) { |
| 45 | // TODO: can NOT expose OSSkey and OSSid in the test. |
| 46 | // Manual test OSS and local file passed. |
| 47 | #if 0 |
| 48 | |
| 49 | NodeDef prefix_def; |
| 50 | Tensor prefix_value(DT_STRING, TensorShape({1})); |
| 51 | prefix_value.flat<std::string>()(0) = |
| 52 | // Local test |
| 53 | //"/local/path/workspace/tmp/DeepFM/ev/1598442950_pai/variables/variables"; |
| 54 | // OSS test |
| 55 | //"oss://bucket-name\x01id=id\x02key=key\x02host=host/jktest/mm/odl_test_files/saved_model/variables"; |
| 56 | TF_CHECK_OK(NodeDefBuilder("prefix", "Const") |
| 57 | .Attr("dtype", DT_STRING) |
| 58 | .Attr("value", prefix_value) |
| 59 | .Finalize(&prefix_def)); |
| 60 | |
| 61 | NodeDef tensor_name_def; |
| 62 | Tensor tensor_name_value(DT_STRING, TensorShape({1})); |
| 63 | tensor_name_value.flat<std::string>()(0) = |
| 64 | "input_from_feature_columns/fm_10169_embedding/weights"; |
| 65 | TF_CHECK_OK(NodeDefBuilder("tensor_name", "Const") |
| 66 | .Attr("dtype", DT_STRING) |
| 67 | .Attr("value", tensor_name_value) |
| 68 | .Finalize(&tensor_name_def)); |
| 69 | |
| 70 | NodeDef storage_pointer_def; |
| 71 | Tensor storage_pointer_value(DT_UINT64, TensorShape({1})); |
| 72 | storage_pointer_value.scalar<tensorflow::uint64>()() = |
| 73 | (uint64_t)(12345678); |
| 74 | TF_CHECK_OK(NodeDefBuilder("storage_pointer", "Const") |
| 75 | .Attr("dtype", DT_UINT64) |
| 76 | .Attr("value", storage_pointer_value) |
| 77 | .Finalize(&storage_pointer_def)); |
| 78 | |
| 79 | NodeDef version_def; |
| 80 | Tensor version_value(DT_UINT64, TensorShape({1})); |
| 81 | version_value.scalar<tensorflow::uint64>()() = |
| 82 | (uint64_t)(1688); |
| 83 | TF_CHECK_OK(NodeDefBuilder("model_version", "Const") |
| 84 | .Attr("dtype", DT_UINT64) |
| 85 | .Attr("value", version_value) |
| 86 | .Finalize(&version_def)); |
| 87 | |
| 88 | NodeDef incr_ckpt_def; |
| 89 | Tensor incr_ckpt_value(DT_BOOL, TensorShape({1})); |
| 90 | incr_ckpt_value.scalar<bool>()() = false; |
| 91 | TF_CHECK_OK(NodeDefBuilder("incr_ckpt", "Const") |
| 92 | .Attr("dtype", DT_BOOL) |
| 93 | .Attr("value", incr_ckpt_value) |
| 94 | .Finalize(&incr_ckpt_def)); |
| 95 | |
| 96 | NodeDef kv_import_def; |
| 97 | TF_CHECK_OK(NodeDefBuilder("kv_import", "KvImport") |
| 98 | .Input("prefix", 0, DT_STRING) |
| 99 | .Input("tensor_name", 0, DT_STRING) |
| 100 | .Input("storage_pointer", 0, DT_UINT64) |
| 101 | .Input("model_version", 0, DT_UINT64) |
nothing calls this directly
no test coverage detected