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

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.cc 87 context_.ws_dims.assign(ws_pd.dims, ws_pd.dims + ws_pd.ndims);
182 context_.ws_dims.assign(ws_pd.dims, ws_pd.dims + ws_pd.ndims);
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},
mkl_maxpooling_op.cc 769 memory::dims ws_dims; variable
770 ws_dims.assign(ws_md.data.dims, ws_md.data.dims + ws_md.data.ndims)
    [all...]
mkl_pooling_ops_common.h 111 memory::dims ws_dims; member in struct:tensorflow::MklPoolingFwdPrimitive::PoolingFwdContext
259 mkldnn::memory::dims ws_dims; member in struct:tensorflow::MklPoolingBwdPrimitive::PoolingBwdContext

Completed in 98 milliseconds