| /aosp_15_r20/external/pytorch/test/distributions/ |
| H A D | test_constraints.py | 85 def build_constraint(constraint_fn, args, is_cuda=False): argument
102 def test_constraint(constraint_fn, result, value, is_cuda): argument
119 def test_biject_to(constraint_fn, args, is_cuda): argument
160 def test_transform_to(constraint_fn, args, is_cuda): argument
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | transformer.cpp | 55 bool is_cuda, in transformer_encoder_layer_test_helper()
247 bool is_cuda, in transformer_decoder_layer_test_helper()
453 bool is_cuda, in transformer_decoder_layer_test_helper_gelu()
566 bool is_cuda, in transformer_encoder_test_helper()
840 bool is_cuda, in transformer_decoder_test_helper()
1366 void transformer_test_helper(bool is_cuda, bool use_callable_activation) { in transformer_test_helper()
|
| H A D | serialize.cpp | 891 bool is_cuda = false) { in TEST()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | AccumulateType.cpp | 27 c10::ScalarType toAccumulateType(c10::ScalarType type, bool is_cuda) { in toAccumulateType()
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | GeneratorImpl.cpp | 90 uint64_t getNonDeterministicRandom(bool is_cuda) { in getNonDeterministicRandom()
|
| H A D | Device.h | 81 bool is_cuda() const noexcept { in is_cuda() function
|
| H A D | TensorImpl.h | 1105 bool is_cuda() const { in is_cuda() function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_foreach.py | 77 def __call__(self, inputs, is_cuda, expect_fastpath, **kwargs): argument
142 def is_cuda(self): member in TestForeach
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/utils/ |
| H A D | rnn.h | 151 bool is_cuda() const { in is_cuda() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | DeprecatedTypeProperties.h | 45 bool is_cuda() const { in is_cuda() function
|
| H A D | interned_strings.cpp | 136 bool Symbol::is_cuda() const { return ns() == namespaces::cuda; } in is_cuda() function in c10::Symbol
|
| H A D | TensorBase.h | 431 bool is_cuda() const { in is_cuda() function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/runtime/ |
| H A D | hints.py | 178 def is_cuda(self): member in HalideMeta
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/_expanded_weights/ |
| H A D | expanded_weights_impl.py | 169 def is_cuda(self): member in ExpandedWeight
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | storage.py | 193 def is_cuda(self): member in _StorageBase
459 def is_cuda(self): member in UntypedStorage
858 def is_cuda(self): member in TypedStorage
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | rnn.py | 215 def is_cuda(self) -> bool: member in PackedSequence
|
| /aosp_15_r20/external/pytorch/torch/csrc/tensor/ |
| H A D | python_tensor.cpp | 37 bool is_cuda; member
|
| /aosp_15_r20/external/pytorch/torch/cuda/ |
| H A D | __init__.py | 1310 is_cuda = True variable in _CudaBase
|