Searched refs:record_temp_memory_allocation (Results 1 – 2 of 2) sorted by relevance
803 record_temp_memory_allocation(alloc_size, *out_temp); in allocate_temp()1042 void OpKernelContext::record_temp_memory_allocation(int64 size, in record_temp_memory_allocation() function in tensorflow::OpKernelContext
1220 void record_temp_memory_allocation(int64 size, const Tensor& t)