Searched refs:report_tensor_allocations_upon_oom (Results 1 – 10 of 10) sorted by relevance
38 name: "report_tensor_allocations_upon_oom"
1471 options.report_tensor_allocations_upon_oom = (1472 options.report_tensor_allocations_upon_oom or1473 incoming_options.report_tensor_allocations_upon_oom)
1394 debug_tensor_watch, report_tensor_allocations_upon_oom): argument1400 report_tensor_allocations_upon_oom)1410 report_tensor_allocations_upon_oom=self1802 report_tensor_allocations_upon_oom=True),1832 report_tensor_allocations_upon_oom=True)1850 report_tensor_allocations_upon_oom=True),
167 bool report_tensor_allocations_upon_oom = false; member
599 if (pss->report_tensor_allocations_upon_oom) { in RunPartitionsHelper()1652 pss.report_tensor_allocations_upon_oom = in DoPartialRun()1653 req.options().report_tensor_allocations_upon_oom(); in DoPartialRun()1789 out_pss->report_tensor_allocations_upon_oom = in FillPerStepState()1790 run_options.report_tensor_allocations_upon_oom(); in FillPerStepState()
180 if (request->exec_opts().report_tensor_allocations_upon_oom() || in DoRunGraph()
667 report_tensor_allocations_upon_oom=True))697 report_tensor_allocations_upon_oom=True))
209 bool report_tensor_allocations_upon_oom = 5; field
488 bool report_tensor_allocations_upon_oom = 7; field
556 run_options.report_tensor_allocations_upon_oom()) { in RunInternal()