Home
last modified time | relevance | path

Searched refs:int_var_map_ (Results 1 – 2 of 2) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/compute/
H A Dkernel_ctx.hpp46 for (auto &int_var : int_var_map_) in options()
57 int_var_map_.insert({variable, value}); in define_int()
78 return int_var_map_.count(name) != 0 || float_var_map_.count(name) != 0; in has_macro()
110 if (int_var_map_.count("DT_F16") != 0) return "f16"; in data_type()
112 if (int_var_map_.count("DT_F32") != 0) return "f32"; in data_type()
114 if (int_var_map_.count("DT_S8") != 0) return "s8"; in data_type()
125 std::map<std::string, int64_t> int_var_map_; member in dnnl::impl::gpu::compute::kernel_ctx_t
132 assert(int_var_map_.count(name) != 0 && "not expected"); in get_scalar()
133 return int_var_map_.at(name); in get_scalar()
/dports/math/onednn/oneDNN-2.5.1/src/gpu/compute/
H A Dkernel_ctx.hpp46 for (auto &int_var : int_var_map_) in options()
57 int_var_map_.insert({variable, value}); in define_int()
78 return int_var_map_.count(name) != 0 || float_var_map_.count(name) != 0; in has_macro()
110 if (int_var_map_.count("DT_F16") != 0) return "f16"; in data_type()
112 if (int_var_map_.count("DT_F32") != 0) return "f32"; in data_type()
114 if (int_var_map_.count("DT_S8") != 0) return "s8"; in data_type()
125 std::map<std::string, int64_t> int_var_map_; member in dnnl::impl::gpu::compute::kernel_ctx_t
132 assert(int_var_map_.count(name) != 0 && "not expected"); in get_scalar()
133 return int_var_map_.at(name); in get_scalar()