Searched refs:record_temp_memory_allocation (Results 1 – 2 of 2) sorted by relevance
848 record_temp_memory_allocation(alloc_size, *out_temp); in allocate_temp()1053 void OpKernelContext::record_temp_memory_allocation(int64_t size, in record_temp_memory_allocation() function in tensorflow::OpKernelContext
1181 void record_temp_memory_allocation(int64_t size, const Tensor& t)