Searched refs:jit_compilation_activity (Results 1 – 5 of 5) sorted by relevance
41 const XlaJitCompilationActivity& jit_compilation_activity) override { in Listen() argument42 jit_compilation_activity_ = jit_compilation_activity; in Listen()55 const XlaJitCompilationActivity& jit_compilation_activity() const { in jit_compilation_activity() function in tensorflow::__anoneb6dd9850111::TestListener164 EXPECT_EQ(listener()->jit_compilation_activity().cluster_name(), "cluster_0"); in TEST_F()165 EXPECT_EQ(listener()->jit_compilation_activity().compile_count(), 1); in TEST_F()168 listener()->jit_compilation_activity().compile_time_us(); in TEST_F()170 EXPECT_EQ(listener()->jit_compilation_activity().cumulative_compile_time_us(), in TEST_F()181 EXPECT_EQ(listener()->jit_compilation_activity().cluster_name(), "cluster_0"); in TEST_F()182 EXPECT_EQ(listener()->jit_compilation_activity().compile_count(), 2); in TEST_F()184 EXPECT_GT(listener()->jit_compilation_activity().compile_time_us(), 0); in TEST_F()[all …]
46 const XlaJitCompilationActivity& jit_compilation_activity) override { in Listen() argument54 VLOG(3) << jit_compilation_activity.DebugString(); in Listen()55 logger->LogProto(jit_compilation_activity); in Listen()
28 Status BroadcastXlaActivity(XlaJitCompilationActivity jit_compilation_activity);54 const XlaJitCompilationActivity& jit_compilation_activity) = 0;
71 XlaJitCompilationActivity jit_compilation_activity) { in BroadcastXlaActivity() argument73 return listener->Listen(jit_compilation_activity); in BroadcastXlaActivity()
484 XlaJitCompilationActivity jit_compilation_activity; in CompileImpl() local485 jit_compilation_activity.set_cluster_name(function.name()); in CompileImpl()486 jit_compilation_activity.set_compile_count(it->second.compile_count); in CompileImpl()487 jit_compilation_activity.set_compile_time_us(compile_time_us); in CompileImpl()488 jit_compilation_activity.set_cumulative_compile_time_us( in CompileImpl()492 BroadcastXlaActivity(std::move(jit_compilation_activity))); in CompileImpl()