Searched refs:run_state_args (Results 1 – 2 of 2) sorted by relevance
747 RunStateArgs run_state_args(run_options.debug_options()); in Run() local748 run_state_args.collective_graph_key = in Run()753 &run_state_args)); in Run()785 LogMemory::RecordStep(step_id, run_state_args.handle); in Run()848 RunStateArgs run_state_args(debug_options); in PRunSetup() local849 run_state_args.is_partial_run = true; in PRunSetup()852 &run_state_args)); in PRunSetup()863 .emplace(run_state_args.handle, in PRunSetup()866 return errors::Internal("The handle '", run_state_args.handle, in PRunSetup()896 LogMemory::RecordStep(args.step_id, run_state_args.handle); in PRunSetup()[all …]
225 ExecutorsAndKeys** executors_and_keys, RunStateArgs* run_state_args);233 RunStateArgs* run_state_args);242 RunStateArgs* run_state_args, DataTypeVector* input_types,