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