Lines Matching refs:pss
177 int64 execution_count, PerStepState* pss,
187 void ProcessStats(int64 step_id, PerStepState* pss, ProfileHandler* ph,
495 PerStepState* pss, CallOptions* call_opts, const RunStepRequestWrapper& req,
513 if (pss->report_tensor_allocations_upon_oom) {
516 if (pss->collect_costs) {
519 if (pss->collect_timeline) {
522 if (pss->collect_rpcs) {
525 if (pss->collect_partition_graphs) {
528 if (pss->collect_costs || pss->collect_timeline) {
529 pss->step_stats.resize(partitions_.size());
646 if (pss->collect_timeline) {
647 pss->step_stats[i].Swap(run_graph_resp->mutable_step_stats());
649 if (pss->collect_costs) {
656 if (pss->collect_partition_graphs) {
737 PerStepState* pss,
741 if (!pss->collect_costs && !pss->collect_timeline) return;
744 if (pss->collect_timeline) {
746 RetrieveLogs(step_id, &pss->rpc_stats);
749 const StepStats& ss = pss->step_stats[i];
757 for (const auto& ds : pss->rpc_stats.dev_stats()) {
760 ph->StepDone(pss->start_micros, pss->end_micros,
765 if (pss->collect_timeline) {
767 step_stats_proto.Swap(&pss->rpc_stats);
769 step_stats_proto.MergeFrom(pss->step_stats[i]);
770 pss->step_stats[i].Clear();
772 pss->step_stats.clear();
1446 PerStepState pss;
1449 pss.collect_timeline =
1451 pss.collect_rpcs = req.options().trace_level() == RunOptions::FULL_TRACE;
1452 pss.report_tensor_allocations_upon_oom =
1462 pss.collect_costs =
1465 pss.collect_partition_graphs = req.options().output_partition_graphs();
1470 pss.collect_timeline = true;
1471 pss.collect_rpcs = ph->should_collect_rpcs();
1474 run_state->pss = std::move(pss);
1522 env_, run_state->step_id, run_state->count, &run_state->pss, opts, req,
1528 run_state->pss.end_micros = Env::Default()->NowMicros();
1532 rcg->ProcessStats(run_state->step_id, &run_state->pss, run_state->ph.get(),
1585 PerStepState pss;
1586 pss.start_micros = Env::Default()->NowMicros();
1613 pss.collect_timeline = req.options().trace_level() == RunOptions::FULL_TRACE;
1614 pss.collect_rpcs = req.options().trace_level() == RunOptions::FULL_TRACE;
1615 pss.report_tensor_allocations_upon_oom =
1623 pss.collect_costs =
1626 pss.collect_partition_graphs = req.options().output_partition_graphs();
1631 pss.collect_timeline = true;
1632 pss.collect_rpcs = ph->should_collect_rpcs();
1635 Status s = rcg->RunPartitions(env_, step_id, count, &pss, opts, req, resp,
1638 pss.end_micros = Env::Default()->NowMicros();
1641 rcg->ProcessStats(step_id, &pss, ph.get(), req.options(),
