Searched refs:engine_kind_ (Results 1 – 10 of 10) sorted by relevance
| /dports/math/onednn/oneDNN-2.5.1/src/sycl/ |
| H A D | sycl_engine.cpp | 26 if (engine_kind_ == engine_kind::cpu) return status::unimplemented; in engine_create()
30 auto dev_type = (engine_kind_ == engine_kind::cpu) in engine_create()
73 engine, engine_kind_, dev, ctx, index); in engine_create()
76 if (engine_kind_ == engine_kind::cpu && !dev.is_cpu() && !dev.is_host()) in engine_create()
78 if (engine_kind_ == engine_kind::gpu && !dev.is_gpu()) in engine_create()
83 (engine_kind_ == engine_kind::cpu) in engine_create()
90 if (engine_kind_ == engine_kind::cpu) return status::unimplemented; in engine_create()
|
| H A D | sycl_engine.hpp | 137 : engine_kind_(engine_kind) { in sycl_engine_factory_t()
138 assert(utils::one_of(engine_kind_, engine_kind::cpu, engine_kind::gpu)); in sycl_engine_factory_t()
143 if (engine_kind_ == engine_kind::cpu) return 0; in count()
145 auto dev_type = (engine_kind_ == engine_kind::cpu) in count()
157 engine_kind_t engine_kind_; member in dnnl::impl::sycl::sycl_engine_factory_t
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/sycl/ |
| H A D | sycl_engine.cpp | 26 if (engine_kind_ == engine_kind::cpu) return status::unimplemented; in engine_create()
30 auto dev_type = (engine_kind_ == engine_kind::cpu) in engine_create()
73 engine, engine_kind_, dev, ctx, index); in engine_create()
76 if (engine_kind_ == engine_kind::cpu && !dev.is_cpu() && !dev.is_host()) in engine_create()
78 if (engine_kind_ == engine_kind::gpu && !dev.is_gpu()) in engine_create()
83 (engine_kind_ == engine_kind::cpu) in engine_create()
90 if (engine_kind_ == engine_kind::cpu) return status::unimplemented; in engine_create()
|
| H A D | sycl_engine.hpp | 140 : engine_kind_(engine_kind) { in sycl_engine_factory_t()
141 assert(utils::one_of(engine_kind_, engine_kind::cpu, engine_kind::gpu)); in sycl_engine_factory_t()
146 if (engine_kind_ == engine_kind::cpu) return 0; in count()
148 auto dev_type = (engine_kind_ == engine_kind::cpu) in count()
160 engine_kind_t engine_kind_; member in dnnl::impl::sycl::sycl_engine_factory_t
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/ |
| H A D | dnnl_memory.hpp | 90 engine_kind_ = rhs.engine_kind_; in operator =()
182 dnnl_engine_kind_t engine_kind() const { return engine_kind_; } in engine_kind()
215 dnnl_engine_kind_t engine_kind_ = dnnl_any_engine; member
237 DNN_SAFE(dnnl_engine_get_kind(engine_, &engine_kind_), CRIT); in initialize()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/ |
| H A D | dnnl_memory.hpp | 90 engine_kind_ = rhs.engine_kind_; in operator =()
182 dnnl_engine_kind_t engine_kind() const { return engine_kind_; } in engine_kind()
215 dnnl_engine_kind_t engine_kind_ = dnnl_any_engine; member
238 DNN_SAFE(dnnl_engine_get_kind(engine_, &engine_kind_), CRIT); in initialize()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | primitive_hashing.hpp | 59 engine_kind_t engine_kind_; member
152 seed, hash_combine(0, static_cast<size_t>(key.engine_kind_))); in operator ()()
|
| H A D | primitive_hashing.cpp | 41 , engine_kind_(engine->kind()) in key_t()
69 && engine_kind_ == rhs.engine_kind_ in operator ==()
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | primitive_hashing.hpp | 59 engine_kind_t engine_kind_; member
152 seed, hash_combine(0, static_cast<size_t>(key.engine_kind_))); in operator ()()
|
| H A D | primitive_hashing.cpp | 41 , engine_kind_(engine->kind()) in key_t()
69 && engine_kind_ == rhs.engine_kind_ in operator ==()
|