Searched refs:set_record_memory_consumption (Results 1 – 3 of 3) sorted by relevance
1205 void set_record_memory_consumption(bool v);
1112 void OpKernelContext::set_record_memory_consumption(bool v) { in set_record_memory_consumption() function in tensorflow::OpKernelContext
669 context->set_record_memory_consumption(true); in Compute()