Searched refs:argument_to_xla_arg (Results 1 – 1 of 1) sorted by relevance
278 absl::flat_hash_map<int, int> argument_to_xla_arg; in BuildComputation() local280 argument_to_xla_arg[input_mapping[xla_arg]] = xla_arg; in BuildComputation()308 alias_resource_update && argument_to_xla_arg.count(param_num)) { in BuildComputation()313 int xla_param_num = argument_to_xla_arg[param_num]; in BuildComputation()