Searched refs:aliased_input (Results 1 – 2 of 2) sorted by relevance
179 [this](const HloInputOutputAliasConfig::Alias& aliased_input) in ComputeInputOutputAliasedBuffers()183 aliased_input.parameter_number), in ComputeInputOutputAliasedBuffers()184 aliased_input.parameter_index); in ComputeInputOutputAliasedBuffers()193 auto aliased_input = in ComputeInputOutputAliasedBuffers() local196 if (aliased_input) { in ComputeInputOutputAliasedBuffers()198 GetBufferForValue(get_parameter_value(*aliased_input))); in ComputeInputOutputAliasedBuffers()
213 int64_t aliased_input = FindAliasedInput(body, i); in HandleWhileOp() local214 if (aliased_input == i) { in HandleWhileOp()319 int64_t aliased_input = FindAliasedInput(info.decomposed_callee, i); in HandlePartitionedCallOp() local320 if (aliased_input >= 0) { in HandlePartitionedCallOp()322 result.replaceAllUsesWith(call.getOperand(aliased_input)); in HandlePartitionedCallOp()