Searched defs:functional_tensor (Results 1 – 2 of 2) sorted by relevance
632 void replace_(const Tensor& functional_tensor, const Tensor& other) { in replace_()637 void replace_(const ITensorListRef functional_tensor, ITensorListRef other) { in replace_()646 void propagate_xla_data(const Tensor& functional_tensor, const Tensor& other) { in propagate_xla_data()654 void propagate_xla_data(const ITensorListRef functional_tensor, ITensorListRef other) { in propagate_xla_data()678 void commit_update(const Tensor& functional_tensor) { in commit_update()683 void commit_update(ITensorListRef functional_tensor) { in commit_update()689 void unsafe_reset_storage(const Tensor& functional_tensor) { in unsafe_reset_storage()694 void mark_mutation_hidden_from_autograd(const Tensor& functional_tensor) { in mark_mutation_hidden_from_autograd()699 bool are_all_mutations_hidden_from_autograd(const Tensor& functional_tensor) { in are_all_mutations_hidden_from_autograd()704 bool are_all_mutations_under_no_grad_or_inference_mode(const Tensor& functional_tensor) { in are_all_mutations_under_no_grad_or_inference_mode()
102 functional_tensor: Optional[FunctionalTensorMetadataEq] = None variable in OutputAliasInfo