Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc530 auto mean_desc = memory::desc({1, bwdParams.depth}, MklDnnType<U>(), in Setup() local
561 context_.mean_mem.reset(new memory(mean_desc, cpu_engine_, DummyData)); in Setup()