Searched defs:aliased_input (Results 1 – 3 of 3) sorted by relevance
425 const at::Tensor& aliased_input = aliased_input_iv.toTensor(); in autogradNotImplementedFallbackImpl() local508 at::Tensor aliased_input; in autogradNotImplementedInplaceOrViewFallbackImpl() local
186 int64_t aliased_input = FindAliasedInput(body, i); in HandleWhileOp() local293 int64_t aliased_input = FindAliasedInput(info.decomposed_callee, i); in HandlePartitionedCallOp() local
61 std::optional<HloInputOutputAliasConfig::Alias> aliased_input = in ComputeInputOutputAliasedValues() local