HomeSort by relevance Sort by last modified time
    Searched refs:ws_fmt (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 108 memory::format ws_fmt; member in struct:tensorflow::MklPoolingFwdPrimitive::PoolingFwdContext
136 ws_fmt(memory::format::any),
243 memory::format GetWorkspaceFormat() const { return context_.ws_fmt; }
256 mkldnn::memory::format ws_fmt; member in struct:tensorflow::MklPoolingBwdPrimitive::PoolingBwdContext
286 ws_fmt(memory::format::any),
mkl_pooling_ops_common.cc 86 context_.ws_fmt = static_cast<mkldnn::memory::format>(ws_pd.format);
183 context_.ws_fmt = get_desired_format(context_.ws_dims[1], is_2d);
186 {{{context_.ws_dims}, context_.ws_dt, context_.ws_fmt}, cpu_engine},

Completed in 523 milliseconds