| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 74 : OpKernel(context), cpu_engine_(engine::kind::cpu, 0) { in MklLRNOp()
124 MklDnnData<T> src_dnn_data(&cpu_engine_); in Compute()
125 MklDnnData<T> dst_dnn_data(&cpu_engine_); in Compute()
126 MklDnnData<uint8> workspace_dnn_data(&cpu_engine_); in Compute()
151 auto lrn_prim_desc = lrn_forward::primitive_desc(lrn_desc, cpu_engine_); in Compute()
167 src_dnn_data.CheckReorderToOpMem(lrn_prim_desc.src_desc(), cpu_engine_); in Compute()
170 fwd_stream_.reset(CreateStream(context, cpu_engine_)); in Compute()
309 engine cpu_engine_; member in tensorflow::MklLRNOp
317 : OpKernel(context), cpu_engine_(engine::kind::cpu, 0) { in MklLRNGradOp()
332 bwd_stream_.reset(new stream(cpu_engine_)); in MklLRNGradOp()
[all …]
|
| D | mkl_matmul_op_fused.cc | 183 auto fuse_add_src_ = memory(add_md, this->cpu_engine_, add_buf); in Compute()
184 auto fuse_add_dst_ = memory(dst_md, this->cpu_engine_, dst_buf); in Compute()
186 ReorderPd(this->cpu_engine_, add_md, this->cpu_engine_, dst_md); in Compute()
189 this->cpu_engine_, ctx); in Compute()
209 MklDnnData<T> src_mkl(&(this->cpu_engine_)); in Compute()
210 MklDnnData<T> weight_mkl(&(this->cpu_engine_)); in Compute()
219 this->cpu_engine_, ctx); in Compute()
246 this->cpu_engine_, ctx); in Compute()
|
| D | mkl_requantize_per_channel_op.cc | 124 new memory(input_md, cpu_engine_, input_buf)); in Compute()
126 new memory(output_md, cpu_engine_, output_buf)); in Compute()
129 ReorderPd(cpu_engine_, input_mem_prim->get_desc(), cpu_engine_, in Compute()
132 reorder_stream.reset(CreateStream(ctx, cpu_engine_)); in Compute()
168 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklRequantizePerChannelOp
|
| D | mkl_fused_batch_norm_op.cc | 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()
216 cpu_engine_, DummyData)); in Setup()
222 cpu_engine_, DummyData)); in Setup()
226 cpu_engine_, DummyData)); in Setup()
231 cpu_engine_, DummyData)); in Setup()
544 auto fwd_pd = BatchNormFwdPd(fwd_desc, cpu_engine_); in Setup()
553 context_.bwd_pd.reset(new BatchNormBwdPd(bwd_desc, cpu_engine_, fwd_pd)); in Setup()
556 context_.src_mem.reset(new memory(src_md, cpu_engine_, DummyData)); in Setup()
[all …]
|
| D | mkl_pooling_ops_common.cc | 58 new pooling_forward::primitive_desc(*context_.fwd_desc, cpu_engine_)); 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()
71 cpu_engine_, DummyData)); in Setup()
159 new pooling_forward::primitive_desc(*context_.fwd_desc, cpu_engine_)); in Setup()
161 *context_.bwd_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
165 cpu_engine_, DummyData)); in Setup()
167 cpu_engine_, DummyData)); in Setup()
172 new memory(context_.fwd_pd.get()->workspace_desc(), cpu_engine_)); in Setup()
|
| D | mkl_maxpooling_op.cc | 64 MklDnnData<T> dnn_data_input(&cpu_engine_); in Compute()
65 MklDnnData<T> dnn_data_output(&cpu_engine_); in Compute()
180 MklDnnData<uint8> dnn_data_wksp(&cpu_engine_); in Compute()
200 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklMaxPoolingOp
250 MklDnnData<T> grad_dnn_data(&cpu_engine_); in Compute()
251 MklDnnData<uint8> workspace_dnn_data(&cpu_engine_); in Compute()
322 cpu_engine_, context); in Compute()
356 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklMaxPoolingGradOp
|
| D | mkl_conv_grad_input_ops.cc | 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()
220 context_.bwd_input_pd.get()->diff_src_desc(), cpu_engine_, DummyData)); in Setup()
222 context_.bwd_input_pd.get()->weights_desc(), cpu_engine_, DummyData)); in Setup()
224 context_.bwd_input_pd.get()->diff_dst_desc(), cpu_engine_, DummyData)); in Setup()
461 MklDnnData<T> filter(&cpu_engine_); in Compute()
465 cpu_engine_, context); in Compute()
473 MklDnnData<T> diff_dst(&cpu_engine_); in Compute()
477 cpu_engine_, context); in Compute()
508 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklConvCustomBackpropInputOp
|
| D | mkl_conv_grad_filter_ops.cc | 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()
260 new memory(bwd_filter_pd->src_desc(), cpu_engine_, DummyData)); in Setup()
262 new memory(bwd_filter_pd->diff_weights_desc(), cpu_engine_, DummyData)); in Setup()
264 new memory(bwd_filter_pd->diff_dst_desc(), cpu_engine_, DummyData)); in Setup()
272 cpu_engine_, DummyData)); in Setup()
539 MklDnnData<T> src(&cpu_engine_); in Compute()
543 src.CheckReorderToOpMem(bwd_filter_pd->src_desc(), cpu_engine_, in Compute()
551 MklDnnData<T> diff_dst(&cpu_engine_); in Compute()
555 cpu_engine_, context); in Compute()
[all …]
|
| D | mkl_matmul_ops_common.h | 199 *context_.fwd_desc, cpu_engine_)); in Setup()
258 *context_.fwd_desc, post_ops_attr, cpu_engine_)); in Setup()
261 *context_.fwd_desc, cpu_engine_)); in Setup()
266 new memory(context_.fwd_pd.get()->src_desc(), cpu_engine_, DummyData)); in Setup()
268 cpu_engine_, DummyData)); in Setup()
270 new memory(context_.fwd_pd.get()->dst_desc(), cpu_engine_, DummyData)); in Setup()
274 cpu_engine_, DummyData)); in Setup()
440 weight.CheckReorderToOpMem(matmul_fwd_pd.get()->weights_desc(), cpu_engine_, in CacheWeight()
491 engine cpu_engine_ = engine(engine::kind::cpu, 0); variable
612 new matmul::primitive_desc(*context_.desc, cpu_engine_)); in Setup()
[all …]
|
| D | mkl_conv_ops.cc | 247 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
280 new ConvFwdPd(*context_.fwd_desc, post_ops_attr, cpu_engine_)); 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()
289 cpu_engine_, DummyData)); in Setup()
291 new memory(context_.fwd_pd.get()->dst_desc(), cpu_engine_, DummyData)); in Setup()
297 cpu_engine_, DummyData)); in Setup()
511 MklDnnData<Tinput> src(&cpu_engine_); in Compute()
512 MklDnnData<Tfilter> filter(&cpu_engine_); in Compute()
683 src.CheckReorderToOpMem(conv_fwd_pd->src_desc(), cpu_engine_, context); in Compute()
[all …]
|
| D | mkl_avgpooling_op.cc | 60 MklDnnData<T> dnn_data_input(&cpu_engine_); in Compute()
166 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklAvgPoolingOp
192 MklDnnData<T> grad_dnn_data(&cpu_engine_); in Compute()
269 cpu_engine_); in Compute()
296 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklAvgPoolingGradOp
|
| D | mkl_qmatmul_op.cc | 174 MklDnnData<Tinput> src(&(this->cpu_engine_)); in Compute()
175 MklDnnData<Tweight> weight(&(this->cpu_engine_)); in Compute()
252 this->cpu_engine_, context); in Compute()
281 this->cpu_engine_, context); in Compute()
479 this->cpu_engine_, bias_buf); in GetBiasHandle()
481 new memory(mkldnn_matmul_fwd_pd->bias_desc(), this->cpu_engine_); in GetBiasHandle()
|
| D | mkl_softmax_op.cc | 124 *context_.fwd_desc, cpu_engine_)); in Setup()
128 new memory(*context_.src_md, cpu_engine_, DummyData)); in Setup()
130 new memory(context_.fwd_pd.get()->dst_desc(), cpu_engine_, DummyData)); in Setup()
|
| D | mkl_quantize_op.cc | 130 new memory(fwdParams.src_md, cpu_engine_, DummyData)); in Setup()
132 new memory(fwdParams.dst_md, cpu_engine_, DummyData)); in Setup()
145 new ReorderPd(cpu_engine_, context_.src_mem->get_desc(), cpu_engine_, in Setup()
|
| D | mkl_relu_op.cc | 145 context_.fwd_pd.reset(new EltwiseFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
150 new memory(fwd_pd->src_desc(), cpu_engine_, DummyData)); in Setup()
152 new memory(fwd_pd->dst_desc(), cpu_engine_, DummyData)); in Setup()
346 context_.fwd_pd.reset(new EltwiseFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
351 new EltwiseBwdPd(*context_.bwd_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
357 new memory(bwd_pd->src_desc(), cpu_engine_, DummyData)); in Setup()
359 new memory(bwd_pd->diff_dst_desc(), cpu_engine_, DummyData)); in Setup()
361 new memory(bwd_pd->diff_src_desc(), cpu_engine_, DummyData)); in Setup()
|
| D | mkl_slice_op.cc | 230 new memory(sliceParams.from->get_desc(), cpu_engine_, DummyData)); in Setup()
232 new memory(sliceParams.to->get_desc(), cpu_engine_, DummyData)); in Setup()
236 context_.src_sub_mem.reset(new memory(src_sub_desc, cpu_engine_, nullptr)); in Setup()
|
| D | mkl_concat_op.cc | 348 new mkldnn::memory(source_md, cpu_engine_, DummyData)); in Setup()
359 cpu_engine_)); in Setup()
363 new memory(*context_.dst_md, cpu_engine_, DummyData)); in Setup()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1264 const engine* cpu_engine_; variable
1273 cpu_engine_(e) {} in MklDnnData()
1279 cpu_engine_ = nullptr; // We don't own this. in ~MklDnnData()
1360 DCHECK(cpu_engine_);
1364 user_memory_ = new memory(pd, *cpu_engine_, data_buffer);
1366 user_memory_ = new memory(pd, *cpu_engine_);
1604 *cpu_engine_, ctx);
1624 reorder_memory_ = new memory(op_pd, *cpu_engine_); in PrepareReorderToUserMemIfReq()
1655 DCHECK(cpu_engine_);
1675 MklPrimitive(const engine& cpu_engine) { cpu_engine_ = cpu_engine; } in MklPrimitive()
[all …]
|