Searched refs:output_mapping (Results 1 – 3 of 3) sorted by relevance
253 std::vector<std::pair<int, int>>* output_mapping);
733 std::vector<std::pair<int, int>>* output_mapping) { in RemoveFunctionOutputs() argument734 DCHECK(output_mapping->empty()); in RemoveFunctionOutputs()765 output_mapping->push_back({i, i - remove_output_args.size()}); in RemoveFunctionOutputs()
264 std::vector<std::pair<int, int>> output_mapping; member369 for (const auto& pair : specialized_func.output_mapping) { in AddTensorMapping()821 std::vector<std::pair<int, int>> output_mapping; in SpecializeFunction() local833 TF_RETURN_IF_ERROR(RemoveFunctionOutputs(remove, &item, &output_mapping)); in SpecializeFunction()863 control_deps, signature.active_outputs, output_mapping}; in SpecializeFunction()