Skip to content

Commit 96d66ab

Browse files
authored
[Embedding] Add performance test for EmbeddingVariable. (#910)
Signed-off-by: lixy9474 <[email protected]>
1 parent 3c4796b commit 96d66ab

File tree

2 files changed

+461
-0
lines changed

2 files changed

+461
-0
lines changed

tensorflow/core/kernels/BUILD

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -460,6 +460,29 @@ tf_cuda_cc_test(
460460
],
461461
)
462462

463+
tf_cuda_cc_test(
464+
name = "embedding_variable_memory_performance_test",
465+
srcs = ["embedding_variable_memory_performance_test.cc"],
466+
extra_copts = ["-fexceptions", "-g"],
467+
deps = [
468+
":io",
469+
":ops_testutil",
470+
":ops_util",
471+
"//tensorflow/core:all_kernels",
472+
"//tensorflow/core:core_cpu",
473+
"//tensorflow/core:core_cpu_internal",
474+
"//tensorflow/core:direct_session_internal",
475+
"//tensorflow/core/util/tensor_bundle",
476+
"//tensorflow/core:framework",
477+
"//tensorflow/core:lib",
478+
"//tensorflow/core:test",
479+
"//tensorflow/core:protos_all_cc",
480+
"//tensorflow/core:testlib",
481+
"//third_party/eigen3",
482+
"//tensorflow/core:test_main",
483+
],
484+
)
485+
463486
cc_library(
464487
name = "tensor_flag_utils",
465488
srcs = [

0 commit comments

Comments
 (0)