Searched refs:report_tensor_allocations_upon_oom (Results 1 – 10 of 10) sorted by relevance
38 name: "report_tensor_allocations_upon_oom"
165 bool report_tensor_allocations_upon_oom = false; member
650 if (pss->report_tensor_allocations_upon_oom) { in RunPartitionsHelper()1744 pss.report_tensor_allocations_upon_oom = in DoPartialRun()1745 req.options().report_tensor_allocations_upon_oom(); in DoPartialRun()1881 out_pss->report_tensor_allocations_upon_oom = in FillPerStepState()1882 run_options.report_tensor_allocations_upon_oom(); in FillPerStepState()
193 if (request->exec_opts().report_tensor_allocations_upon_oom() || in DoRunGraph()
1514 options.report_tensor_allocations_upon_oom = (1515 options.report_tensor_allocations_upon_oom or1516 incoming_options.report_tensor_allocations_upon_oom)
1440 debug_tensor_watch, report_tensor_allocations_upon_oom): argument1446 report_tensor_allocations_upon_oom)1456 report_tensor_allocations_upon_oom=self1848 report_tensor_allocations_upon_oom=True),1878 report_tensor_allocations_upon_oom=True)1896 report_tensor_allocations_upon_oom=True),
666 report_tensor_allocations_upon_oom=True))697 report_tensor_allocations_upon_oom=True))
226 bool report_tensor_allocations_upon_oom = 5; field
675 bool report_tensor_allocations_upon_oom = 7; field
670 run_options.report_tensor_allocations_upon_oom()) { in RunInternal()