| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_grad_filter_ops.cc | 177 std::shared_ptr<mkldnn::convolution_forward::primitive_desc> fwd_pd; member in struct:tensorflow::MklConvBwdFilterPrimitive::ConvBwdFilterContext
199 fwd_pd(nullptr),
249 context_.fwd_pd.reset(new convolution_forward::primitive_desc(
255 *context_.bwd_filter_desc, cpu_engine_, *context_.fwd_pd));
|
| mkl_conv_grad_input_ops.cc | 147 std::shared_ptr<mkldnn::convolution_forward::primitive_desc> fwd_pd; member in struct:tensorflow::MklConvBwdInputPrimitive::ConvBwdInputContext
168 fwd_pd(nullptr),
199 context_.fwd_pd.reset(new convolution_forward::primitive_desc(
204 *context_.bwd_input_desc, cpu_engine_, *context_.fwd_pd));
|
| mkl_pooling_ops_common.h | 88 return context_.fwd_pd;
122 std::shared_ptr<mkldnn::pooling_forward::primitive_desc> fwd_pd; member in struct:tensorflow::MklPoolingFwdPrimitive::PoolingFwdContext
141 fwd_pd(nullptr),
231 return context_.fwd_pd;
274 std::shared_ptr<mkldnn::pooling_forward::primitive_desc> fwd_pd; member in struct:tensorflow::MklPoolingBwdPrimitive::PoolingBwdContext
294 fwd_pd(nullptr),
|
| mkl_relu_op.cc | 91 return context_.fwd_pd;
108 std::shared_ptr<mkldnn::eltwise_forward::primitive_desc> fwd_pd; member in struct:tensorflow::MklEltwiseFwdPrimitive::EltwiseFwdContext
128 fwd_pd(nullptr),
147 context_.fwd_pd.reset(new mkldnn::eltwise_forward::primitive_desc(
153 new memory(context_.fwd_pd.get()->dst_primitive_desc(), DummyData));
157 *context_.fwd_pd, *context_.src_mem, *context_.dst_mem));
312 std::shared_ptr<mkldnn::eltwise_forward::primitive_desc> fwd_pd; member in struct:tensorflow::MklEltwiseBwdPrimitive::EltwiseBwdContext
333 fwd_pd(nullptr),
354 context_.fwd_pd.reset(new mkldnn::eltwise_forward::primitive_desc(
360 *context_.bwd_desc, cpu_engine_, *context_.fwd_pd));
[all...] |
| mkl_fused_batch_norm_op.cc | 153 auto fwd_pd = local
158 context_.dst_mem.reset(new memory(fwd_pd.dst_primitive_desc(), DummyData));
182 fwd_pd, *context_.src_mem, *context_.weights_mem,
186 fwd_pd, *context_.src_mem, *context_.dst_mem));
191 fwd_pd, *context_.src_mem, (const primitive::at)*context_.mean_mem,
196 fwd_pd, *context_.src_mem, (const primitive::at)*context_.mean_mem,
202 fwd_pd, *context_.src_mem, *context_.weights_mem, *context_.dst_mem,
206 fwd_pd, *context_.src_mem, *context_.dst_mem, *context_.mean_mem,
406 auto fwd_pd = local
420 bwd_desc, cpu_engine_, fwd_pd);
[all...] |
| mkl_conv_ops.cc | 166 return context_.fwd_pd;
192 std::shared_ptr<ConvFwdPd> fwd_pd; member in struct:tensorflow::MklConvFwdPrimitive::ConvFwdContext
209 fwd_pd(nullptr),
244 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_));
275 context_.fwd_pd.reset(
278 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_));
283 context_.fwd_pd.get()->src_primitive_desc().desc().data.format);
286 context_.fwd_pd.get()->weights_primitive_desc().desc().data.format);
290 new memory(context_.fwd_pd.get()->src_primitive_desc(), DummyData));
292 new memory(context_.fwd_pd.get()->weights_primitive_desc(), DummyData))
[all...] |
