Searched defs:BatchNormBwdContext (Results 1 – 1 of 1) sorted by relevance
487 struct BatchNormBwdContext { struct in tensorflow::MklFusedBatchNormBwdPrimitive489 int64 flags;492 std::shared_ptr<mkldnn::memory> src_mem;493 std::shared_ptr<mkldnn::memory> mean_mem;494 std::shared_ptr<mkldnn::memory> variance_mem;495 std::shared_ptr<mkldnn::memory> diff_dst_mem;496 std::shared_ptr<mkldnn::memory> weights_mem;497 std::shared_ptr<mkldnn::memory> diff_weights_mem;498 std::shared_ptr<mkldnn::memory> diff_src_mem;501 std::shared_ptr<BatchNormBwdPd> bwd_pd;[all …]