Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_program_group_interface.h54 virtual bool LogProgramMemorySummary() = 0;
H A Dtpu_program_group.h117 bool LogProgramMemorySummary() override;
H A Dtpu_program_group.cc136 bool TpuProgramGroup::LogProgramMemorySummary() { in LogProgramMemorySummary() function in tensorflow::tpu::TpuProgramGroup
H A Dtpu_compile_op_common.cc213 tpu_program_group->LogProgramMemorySummary(); in CompileLocallyAndFillHostCacheInternal()
H A Dtpu_compilation_cache_interface.cc478 entry->tpu_program_group->LogProgramMemorySummary(); in CompileIfKeyAbsentHelper()