Home
last modified time | relevance | path

Searched refs:LogProgramMemorySummary (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_program_group_interface.h52 virtual bool LogProgramMemorySummary() = 0;
Dtpu_program_group.h117 bool LogProgramMemorySummary() override;
Dtpu_program_group.cc133 bool TpuProgramGroup::LogProgramMemorySummary() { in LogProgramMemorySummary() function in tensorflow::tpu::TpuProgramGroup
Dtpu_compilation_cache_interface.cc474 entry->tpu_program_group->LogProgramMemorySummary(); in CompileIfKeyAbsentHelper()
Dtpu_compile_op_common.cc621 tpu_program_group->LogProgramMemorySummary(); in CompileLocallyAndFillHostCache()