Home
last modified time | relevance | path

Searched full:device_check (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dnative_functions.yaml220 …device_check: NoCheck # Tensor arguments allowed to be on different devices, see also _cudnn_ctc_…
225 …device_check: NoCheck # Tensor arguments allowed to be on different devices, see also _cudnn_ctc_…
230 …device_check: NoCheck # log_probs is expected to be on CUDA while targets is expected to be on CPU
236 …device_check: NoCheck # log_probs is expected to be on CUDA while targets is expected to be on CPU
250 device_check: NoCheck
336 device_check: NoCheck # TensorIterator
346 device_check: NoCheck # TensorIterator
355 device_check: NoCheck # TensorIterator
388 device_check: NoCheck # TensorIterator
392 device_check: NoCheck # TensorIterator
[all …]
H A DREADME.md437 ### `device_check`
440 device_check: NoCheck
450 `device_check: NoCheck` to your function definition.
H A DCPUBlas.cpp1077 static inline bool device_check(ScalarType dtype) { in device_check() function
1142 if (Brgemm::device_check(ScalarType::Half)) { in brgemm()
/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_executorch_gen.py30 device_check: NoCheck # TensorIterator
45 device_check: NoCheck # TensorIterator
57 device_check: NoCheck # TensorIterator
69 device_check: NoCheck # TensorIterator
85 device_check: NoCheck # TensorIterator
112 device_check: NoCheck # TensorIterator
118 device_check: NoCheck # TensorIterator
136 device_check: NoCheck # TensorIterator
164 device_check: NoCheck # TensorIterator
169 device_check: NoCheck # TensorIterator
[all …]
/aosp_15_r20/external/executorch/codegen/tools/test/
H A Dtest_gen_oplist.py26 device_check: NoCheck # TensorIterator
31 device_check: NoCheck # TensorIterator
201 device_check: NoCheck # TensorIterator
206 device_check: NoCheck # TensorIterator
220 device_check: NoCheck # TensorIterator
225 device_check: NoCheck # TensorIterator
/aosp_15_r20/external/pytorch/torchgen/dest/
H A Dregister_dispatch_key.py275 device_check = "std::optional<Device> common_device = std::nullopt;\n"
276 device_check += "(void)common_device; // Suppress unused variable warning\n"
280 device_check += f"""
282 return device_check
495 device_check = " // No device check\n"
501 device_check = RegisterDispatchKey.gen_device_check(
502 f.device_check, list(device_check_args), name
553 {device_check}
846 f.device_check, list(device_check_args), sig.name()
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py495 device_check: DeviceCheckType
643 device_check_s = e.pop("device_check", None)
649 ), f"illegal device_check: {device_check_s}"
650 device_check: DeviceCheckType
652 device_check = DeviceCheckType.ExactSame
654 device_check = DeviceCheckType[device_check_s]
941 device_check=device_check,
1013 assert self.device_check == DeviceCheckType.NoCheck, (
1015 "device_check not allowed to be enabled"
H A Dnative_function_generation.py354 device_check=DeviceCheckType.NoCheck,