Searched refs:out_pss (Results 1 – 2 of 2) sorted by relevance
1784 PerStepState* out_pss, in FillPerStepState() argument1786 out_pss->collect_timeline = in FillPerStepState()1788 out_pss->collect_rpcs = run_options.trace_level() == RunOptions::FULL_TRACE; in FillPerStepState()1789 out_pss->report_tensor_allocations_upon_oom = in FillPerStepState()1797 out_pss->collect_costs = in FillPerStepState()1800 out_pss->collect_partition_graphs = run_options.output_partition_graphs(); in FillPerStepState()1804 out_pss->collect_timeline = true; in FillPerStepState()1805 out_pss->collect_rpcs = (*out_ph)->should_collect_rpcs(); in FillPerStepState()
232 int64 count, PerStepState* out_pss,