Searched defs:BatchNormFwdContext (Results 1 – 1 of 1) sorted by relevance
149 struct BatchNormFwdContext { struct in tensorflow::MklFusedBatchNormFwdPrimitive151 int64 flags;154 mkldnn::prop_kind pkind;157 std::shared_ptr<mkldnn::memory> src_mem;158 std::shared_ptr<mkldnn::memory> weights_mem;159 std::shared_ptr<mkldnn::memory> dst_mem;160 std::shared_ptr<mkldnn::memory> mean_mem;161 std::shared_ptr<mkldnn::memory> variance_mem;162 std::shared_ptr<mkldnn::memory> ws_mem;165 std::shared_ptr<BatchNormFwdPd> fwd_pd;[all …]