Searched refs:run_state_args (Results 1 – 2 of 2) sorted by relevance
852 RunStateArgs run_state_args(run_options.debug_options()); in Run() local853 run_state_args.collective_graph_key = in Run()858 &run_state_args)); in Run()890 LogMemory::RecordStep(step_id, run_state_args.handle); in Run()954 RunStateArgs run_state_args(debug_options); in PRunSetup() local955 run_state_args.is_partial_run = true; in PRunSetup()958 &run_state_args)); in PRunSetup()969 .emplace(run_state_args.handle, in PRunSetup()972 return errors::Internal("The handle '", run_state_args.handle, in PRunSetup()1002 LogMemory::RecordStep(args.step_id, run_state_args.handle); in PRunSetup()[all …]
246 ExecutorsAndKeys** executors_and_keys, RunStateArgs* run_state_args);254 RunStateArgs* run_state_args);263 RunStateArgs* run_state_args, DataTypeVector* input_types,