Searched refs:remaining_params (Results 1 – 1 of 1) sorted by relevance
589 std::vector<int>* remaining_params, in CompileGraphSetup() argument591 TF_ASSIGN_OR_RETURN(*remaining_params, RewriteWithArgs(module_op, args)); in CompileGraphSetup()592 arg_shapes.reserve(remaining_params->size()); in CompileGraphSetup()593 for (unsigned idx : *remaining_params) { in CompileGraphSetup()623 std::vector<int> remaining_params; in BuildHloFromModule() local626 CompileGraphSetup(module_op, args, &remaining_params, arg_shapes)); in BuildHloFromModule()638 std::vector<int> remaining_params; in CompileGraphToXlaHlo() local641 CompileGraphSetup(module_op, args, &remaining_params, arg_shapes)); in CompileGraphToXlaHlo()647 compilation_result->input_mapping = remaining_params; in CompileGraphToXlaHlo()