Home
last modified time | relevance | path

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

/external/executorch/runtime/executor/test/
Dkernel_integration_test.cpp70 int temp_memory_size = 0; member
81 temp_memory_size = 0; in reset()
141 context.allocate_temp(control->temp_memory_size); in kernel_hook()
143 control->total_allocated_size += control->temp_memory_size; in kernel_hook()
147 for (int i = 0; i < control->temp_memory_size; i++) { in kernel_hook()
160 context.allocate_temp(control->temp_memory_size); in kernel_hook()
161 control->total_allocated_size += control->temp_memory_size; in kernel_hook()
328 control_->temp_memory_size = 4; in TEST_F()
336 control_->temp_memory_size = 8; in TEST_F()
360 control_->temp_memory_size = 4; in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dcostmodel.h216 MemUsage() : temp_memory_size(0), persistent_memory_size(0) {} in MemUsage()
219 Bytes temp_memory_size; member
Dcostmodel.cc312 return max_mem_usage_[id].temp_memory_size; in TempMemorySize()
327 max_mem_usage_[id].temp_memory_size = memory_stats.temp_memory_size(); in RecordMemoryStats()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.cc138 step_stat.memory_stats().temp_memory_size()); in AddMemoryStats()
145 step_stat.memory_stats().temp_memory_size()); in AddMemoryStats()
/external/tensorflow/tensorflow/core/framework/
Dstep_stats.proto45 int64 temp_memory_size = 1; field