| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | SparseTensorImpl.cpp | 9 DeviceType sparseTensorSetToDeviceType(DispatchKeySet key_set) { in sparseTensorSetToDeviceType()
30 SparseTensorImpl::SparseTensorImpl(at::DispatchKeySet key_set, const caffe2::TypeMeta data_type) in SparseTensorImpl()
35 SparseTensorImpl::SparseTensorImpl(at::DispatchKeySet key_set, const caffe2::TypeMeta data_type, at… in SparseTensorImpl()
|
| H A D | SparseCsrTensorImpl.cpp | 12 at::DispatchKeySet key_set, in SparseCsrTensorImpl()
41 at::DispatchKeySet key_set, in SparseCsrTensorImpl()
|
| H A D | TensorSubclassLikeUtils.h | 53 auto key_set = tensor.unsafeGetTensorImpl()->key_set(); in isTensorSubclassLike() local
|
| H A D | NestedTensorImpl.cpp | 167 c10::DispatchKeySet key_set, in NestedTensorImpl()
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | TensorImpl.cpp | 81 DispatchKeySet key_set, in TensorImpl()
111 DispatchKeySet key_set, in TensorImpl()
128 DispatchKeySet key_set, in TensorImpl()
136 DispatchKeySet key_set, in TensorImpl()
196 auto key_set = in _change_backend_component_keys() local
|
| H A D | GeneratorImpl.h | 84 DispatchKeySet key_set() const { in key_set() function
|
| H A D | GeneratorImpl.cpp | 20 GeneratorImpl::GeneratorImpl(Device device_in, DispatchKeySet key_set) in GeneratorImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | test_utils_test.cc | 143 absl::flat_hash_set<uint32_t> key_set; in XLA_TEST_F() local
174 absl::flat_hash_set<int32_t> key_set; in XLA_TEST_F() local
205 absl::flat_hash_set<uint16_t> key_set; in XLA_TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | TensorWrapper.cpp | 59 auto key_set = getKeysToPropagateToWrapper(tensor, keys_to_propagate); in makeTensorWrapperPtr() local
79 auto key_set = getKeysToPropagateToWrapper(tensor, keys_to_propagate); in unsafeMakeTensorWrapper() local
141 c10::DispatchKeySet key_set, in TensorWrapper()
|
| H A D | BatchedTensorImpl.cpp | 15 BatchedTensorImpl::BatchedTensorImpl(DispatchKeySet key_set, Tensor value, int64_t bdim, int64_t le… in BatchedTensorImpl()
175 DispatchKeySet key_set = getKeysToPropagateToWrapper(tensor); in makeBatched() local
|
| H A D | BatchedTensorImpl.h | 165 auto key_set = tensor.unsafeGetTensorImpl()->key_set(); variable
|
| H A D | DynamicLayer.cpp | 469 restoreLocalDispatchKeySetRAII(const c10::impl::LocalDispatchKeySet& key_set) { in restoreLocalDispatchKeySetRAII()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/quantized/ |
| H A D | QTensorImpl.cpp | 7 DispatchKeySet key_set, in QTensorImpl()
16 DispatchKeySet key_set, in QTensorImpl()
|
| /aosp_15_r20/external/googleapis/google/spanner/v1/ |
| H A D | spanner.proto | 1018 KeySet key_set = 6 [(google.api.field_behavior) = REQUIRED]; field
1087 KeySet key_set = 6 [(google.api.field_behavior) = REQUIRED]; field
|
| H A D | mutation.proto | 70 KeySet key_set = 2 [(google.api.field_behavior) = REQUIRED]; field
|
| /aosp_15_r20/external/cronet/net/quic/ |
| H A D | quic_session_key_unittest.cc | 164 std::set<QuicSessionKey> key_set(session_keys.begin(), session_keys.end()); in TEST() local
|
| /aosp_15_r20/external/cronet/net/spdy/ |
| H A D | spdy_session_key_unittest.cc | 158 std::set<SpdySessionKey> key_set(session_keys.begin(), session_keys.end()); in TEST() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | TorchDispatchUtils.cpp | 7 DispatchKeySet key_set({DispatchKey::Python, DispatchKey::PythonTLSSnapshot}); in tensor_has_dispatch() local
|
| H A D | Generator.h | 118 DispatchKeySet key_set() const { in key_set() function
|
| H A D | TensorBase.h | 336 DispatchKeySet key_set() const { in key_set() function
|
| /aosp_15_r20/external/pytorch/c10/core/impl/ |
| H A D | LocalDispatchKeySet.cpp | 22 void _force_tls_local_dispatch_key_set(LocalDispatchKeySet key_set) { in _force_tls_local_dispatch_key_set()
|
| /aosp_15_r20/external/harfbuzz_ng/perf/ |
| H A D | benchmark-map.cc | 76 hb_set_t* key_set = hb_set_create (); in BM_MapLookupHit() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ComplexHelper.h | 25 auto key_set = tensor.key_set().remove(DispatchKey::Conjugate); in view_tensor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| H A D | cupti_collector.cc | 261 absl::flat_hash_set<absl::string_view> key_set; in CreateXEvent() local
|
| H A D | device_tracer_rocm.cc | 812 absl::flat_hash_set<absl::string_view> key_set; in CreateXEvent() local
|