Searched defs:set_tensor (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/nn/utils/ | ||
| H A D | _named_member_accessor.py | 12 def set_tensor(module: "torch.nn.Module", name: str, tensor: torch.Tensor) -> None: function |
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ | ||
| H A D | CUDASparseDescriptors.h | 214 void set_tensor(const Tensor& input) { in set_tensor() function |
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ | ||
| H A D | tensor_impl.cpp | 91 void LTCTensorImpl::set_tensor(const LazyTensorPtr& lazy_tensor) { in set_tensor() function in torch::lazy::LTCTensorImpl |
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/ | ||
| H A D | interpreter.py | 681 def set_tensor(self, tensor_index, value): member in Interpreter |
| /aosp_15_r20/external/pytorch/test/export/ | ||
| H A D | test_export.py | 828 def set_tensor(self, i, val): member in TestExport.test_state_tensors.M |