| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.cc | 57 context_.fwd_pd.reset( in Setup()
63 new memory(context_.fwd_pd.get()->src_desc(), cpu_engine_, DummyData)); in Setup()
65 new memory(context_.fwd_pd.get()->dst_desc(), cpu_engine_, DummyData)); in Setup()
70 context_.ws_mem.reset(new memory(context_.fwd_pd.get()->workspace_desc(), in Setup()
75 context_.fwd.reset(new pooling_forward(*context_.fwd_pd)); in Setup()
79 context_.fwd.reset(new pooling_forward(*context_.fwd_pd)); in Setup()
158 context_.fwd_pd.reset( in Setup()
161 *context_.bwd_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
172 new memory(context_.fwd_pd.get()->workspace_desc(), cpu_engine_)); in Setup()
|
| D | mkl_softmax_op.cc | 82 return context_.fwd_pd; in GetSoftmaxFwdPd()
98 std::shared_ptr<mkldnn::softmax_forward::primitive_desc> fwd_pd; member
109 fwd_pd(nullptr), in SoftmaxFwdContext()
123 context_.fwd_pd.reset(new mkldnn::softmax_forward::primitive_desc( in Setup()
130 new memory(context_.fwd_pd.get()->dst_desc(), cpu_engine_, DummyData)); in Setup()
133 context_.softmax_fwd.reset(new mkldnn::softmax_forward(*context_.fwd_pd)); in Setup()
|
| D | mkl_matmul_ops_common.h | 134 return context_.fwd_pd; in GetPrimitiveDesc()
148 std::shared_ptr<mkldnn::inner_product_forward::primitive_desc> fwd_pd; member
168 fwd_pd(nullptr), in MklDnnMatMulFwdContext()
198 context_.fwd_pd.reset(new inner_product_forward::primitive_desc( in Setup()
257 context_.fwd_pd.reset(new inner_product_forward::primitive_desc( in Setup()
260 context_.fwd_pd.reset(new inner_product_forward::primitive_desc( in Setup()
266 new memory(context_.fwd_pd.get()->src_desc(), cpu_engine_, DummyData)); in Setup()
267 context_.weight_mem.reset(new memory(context_.fwd_pd.get()->weights_desc(), in Setup()
270 new memory(context_.fwd_pd.get()->dst_desc(), cpu_engine_, DummyData)); in Setup()
277 context_.matmul_fwd.reset(new inner_product_forward(*context_.fwd_pd)); in Setup()
|
| D | mkl_relu_op.cc | 97 std::shared_ptr<EltwiseFwdPd> GetEltwiseFwdPd() { return context_.fwd_pd; } in GetEltwiseFwdPd()
108 std::shared_ptr<EltwiseFwdPd> fwd_pd; member
128 fwd_pd(nullptr), in EltwiseFwdContext()
145 context_.fwd_pd.reset(new EltwiseFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
146 auto fwd_pd = context_.fwd_pd.get(); in Setup() local
150 new memory(fwd_pd->src_desc(), cpu_engine_, DummyData)); in Setup()
152 new memory(fwd_pd->dst_desc(), cpu_engine_, DummyData)); in Setup()
154 context_.eltwise_fwd.reset(new eltwise_forward(*context_.fwd_pd)); in Setup()
309 std::shared_ptr<EltwiseFwdPd> fwd_pd; member
329 fwd_pd(nullptr), in EltwiseBwdContext()
[all …]
|
| D | mkl_pooling_ops_common.h | 90 return context_.fwd_pd; in GetPoolingFwdPd()
123 std::shared_ptr<PoolingFwdPd> fwd_pd; member
144 fwd_pd(nullptr), in PoolingFwdContext()
230 return context_.fwd_pd; in GetPoolingFwdPd()
269 std::shared_ptr<PoolingFwdPd> fwd_pd; member
290 fwd_pd(nullptr), in PoolingBwdContext()
|
| D | mkl_conv_grad_input_ops.cc | 152 std::shared_ptr<ConvFwdPd> fwd_pd; member
173 fwd_pd(nullptr), in ConvBwdInputContext()
214 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
216 *context_.bwd_input_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
|
| D | mkl_fused_batch_norm_op.cc | 144 return context_.fwd_pd; in GetBatchNormFwdPd()
165 std::shared_ptr<BatchNormFwdPd> fwd_pd; member
203 context_.fwd_pd.reset(new BatchNormFwdPd(fwd_desc, cpu_engine_)); in Setup()
207 new memory(context_.fwd_pd->src_desc(), cpu_engine_, DummyData)); in Setup()
209 new memory(context_.fwd_pd->dst_desc(), cpu_engine_, DummyData)); in Setup()
230 context_.ws_mem.reset(new memory(context_.fwd_pd->workspace_desc(), in Setup()
246 context_.bn_fwd.reset(new batch_normalization_forward(*context_.fwd_pd)); in Setup()
281 context_.bn_fwd.reset(new batch_normalization_forward(*context_.fwd_pd)); in Setup()
316 context_.bn_fwd.reset(new batch_normalization_forward(*context_.fwd_pd)); in Setup()
544 auto fwd_pd = BatchNormFwdPd(fwd_desc, cpu_engine_); in Setup() local
[all …]
|
| D | mkl_conv_grad_filter_ops.cc | 173 std::shared_ptr<ConvFwdPd> fwd_pd; member
196 fwd_pd(nullptr), in ConvBwdFilterContext()
236 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
254 *context_.bwd_filter_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
|
| D | mkl_conv_ops.cc | 168 return context_.fwd_pd; in GetPrimitiveDesc()
190 std::shared_ptr<ConvFwdPd> fwd_pd; member
205 fwd_pd(nullptr), in ConvFwdContext()
247 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
279 context_.fwd_pd.reset( in Setup()
282 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
287 new memory(context_.fwd_pd.get()->src_desc(), cpu_engine_, DummyData)); in Setup()
288 context_.filter_mem.reset(new memory(context_.fwd_pd.get()->weights_desc(), in Setup()
291 new memory(context_.fwd_pd.get()->dst_desc(), cpu_engine_, DummyData)); in Setup()
298 context_.conv_fwd.reset(new convolution_forward(*context_.fwd_pd)); in Setup()
[all …]
|
| D | mkl_concat_op.cc | 328 std::shared_ptr<mkldnn::concat::primitive_desc> fwd_pd; member
336 : dst_mem(nullptr), fwd_pd(nullptr), concat_fwd(nullptr) {} in ConcatFwdContext()
357 context_.fwd_pd.reset(new concat::primitive_desc( in Setup()
365 context_.concat_fwd.reset(new concat(*context_.fwd_pd)); in Setup()
|