| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_pooling_ops_common.cc | 67 context_.fwd_pd.reset( in Setup()
72 context_.fwd_pd.get()->dst_primitive_desc().desc().data.format); in Setup()
79 new memory(context_.fwd_pd.get()->dst_primitive_desc(), DummyData)); in Setup()
84 auto ws_pd = context_.fwd_pd.get()->workspace_primitive_desc().desc().data; in Setup()
90 context_.fwd_pd.get()->workspace_primitive_desc().get_size(); in Setup()
92 context_.fwd_pd.get()->workspace_primitive_desc(), DummyData)); in Setup()
93 context_.fwd.reset(new pooling_forward(*context_.fwd_pd, *context_.src_mem, in Setup()
97 context_.fwd.reset(new pooling_forward(*context_.fwd_pd, *context_.src_mem, in Setup()
161 context_.fwd_pd.reset( in Setup()
164 *context_.bwd_desc, cpu_engine, *context_.fwd_pd)); in Setup()
[all …]
|
| D | mkl_fused_batch_norm_op.cc | 153 auto fwd_pd = in Setup() local
158 context_.dst_mem.reset(new memory(fwd_pd.dst_primitive_desc(), DummyData)); in Setup()
182 fwd_pd, *context_.src_mem, *context_.weights_mem, in Setup()
186 fwd_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
191 fwd_pd, *context_.src_mem, (const primitive::at)*context_.mean_mem, in Setup()
196 fwd_pd, *context_.src_mem, (const primitive::at)*context_.mean_mem, in Setup()
202 fwd_pd, *context_.src_mem, *context_.weights_mem, *context_.dst_mem, in Setup()
206 fwd_pd, *context_.src_mem, *context_.dst_mem, *context_.mean_mem, in Setup()
406 auto fwd_pd = in Setup() local
420 bwd_desc, cpu_engine_, fwd_pd); in Setup()
|
| D | mkl_relu_op.cc | 91 return context_.fwd_pd; in GetEltwiseFwdPd()
108 std::shared_ptr<mkldnn::eltwise_forward::primitive_desc> fwd_pd; member
128 fwd_pd(nullptr), in EltwiseFwdContext()
147 context_.fwd_pd.reset(new mkldnn::eltwise_forward::primitive_desc( in Setup()
153 new memory(context_.fwd_pd.get()->dst_primitive_desc(), DummyData)); in Setup()
157 *context_.fwd_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
312 std::shared_ptr<mkldnn::eltwise_forward::primitive_desc> fwd_pd; member
333 fwd_pd(nullptr), in EltwiseBwdContext()
354 context_.fwd_pd.reset(new mkldnn::eltwise_forward::primitive_desc( in Setup()
360 *context_.bwd_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
|
| D | mkl_pooling_ops_common.h | 88 return context_.fwd_pd; in GetPoolingFwdPd()
122 std::shared_ptr<mkldnn::pooling_forward::primitive_desc> fwd_pd; member
141 fwd_pd(nullptr), in PoolingFwdContext()
231 return context_.fwd_pd; in GetPoolingFwdPd()
274 std::shared_ptr<mkldnn::pooling_forward::primitive_desc> fwd_pd; member
294 fwd_pd(nullptr), in PoolingBwdContext()
|
| D | mkl_conv_grad_input_ops.cc | 147 std::shared_ptr<mkldnn::convolution_forward::primitive_desc> fwd_pd; member
168 fwd_pd(nullptr), in ConvBwdInputContext()
199 context_.fwd_pd.reset(new convolution_forward::primitive_desc( in Setup()
204 *context_.bwd_input_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
|
| D | mkl_conv_ops.cc | 166 return context_.fwd_pd; in GetPrimitiveDesc()
192 std::shared_ptr<ConvFwdPd> fwd_pd; member
209 fwd_pd(nullptr), in ConvFwdContext()
244 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
275 context_.fwd_pd.reset( in Setup()
278 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
283 context_.fwd_pd.get()->src_primitive_desc().desc().data.format); in Setup()
286 context_.fwd_pd.get()->weights_primitive_desc().desc().data.format); in Setup()
290 new memory(context_.fwd_pd.get()->src_primitive_desc(), DummyData)); in Setup()
292 new memory(context_.fwd_pd.get()->weights_primitive_desc(), DummyData)); in Setup()
[all …]
|
| D | mkl_conv_grad_filter_ops.cc | 177 std::shared_ptr<mkldnn::convolution_forward::primitive_desc> fwd_pd; member
199 fwd_pd(nullptr), in ConvBwdFilterContext()
249 context_.fwd_pd.reset(new convolution_forward::primitive_desc( in Setup()
255 *context_.bwd_filter_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
|