| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_fused_batch_norm_op.cc | 50 : cpu_engine_(engine::cpu, 0) { in MklFusedBatchNormFwdPrimitive()
154 batch_normalization_forward::primitive_desc(fwd_desc, cpu_engine_); in Setup()
157 context_.src_mem.reset(new memory({src_md, cpu_engine_}, DummyData)); in Setup()
164 new memory({weights_desc, cpu_engine_}, DummyData)); in Setup()
170 context_.mean_mem.reset(new memory({mean_desc, cpu_engine_}, DummyData)); in Setup()
175 new memory({variance_desc, cpu_engine_}, DummyData)); in Setup()
219 engine cpu_engine_; member in tensorflow::MklFusedBatchNormFwdPrimitive
291 : cpu_engine_(engine::cpu, 0) { in MklFusedBatchNormBwdPrimitive()
407 batch_normalization_forward::primitive_desc(fwd_desc, cpu_engine_); in Setup()
420 bwd_desc, cpu_engine_, fwd_pd); in Setup()
[all …]
|
| D | mkl_conv_grad_filter_ops.cc | 83 : cpu_engine_(engine::cpu, 0) { in MklConvBwdFilterPrimitive()
250 *context_.fwd_desc, cpu_engine_)); in Setup()
255 *context_.bwd_filter_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
280 cpu_engine_}, in Setup()
295 engine cpu_engine_; member in tensorflow::MklConvBwdFilterPrimitive
371 MklDnnData<T> src(&cpu_engine_); in Compute()
372 MklDnnData<T> diff_dst(&cpu_engine_); in Compute()
373 MklDnnData<T> diff_filter(&cpu_engine_); // output in Compute()
608 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklConvCustomBackpropFilterOp
|
| D | mkl_requantize_per_channel_op.cc | 108 memory::primitive_desc(input_md, cpu_engine_); in Compute()
110 memory::primitive_desc(output_md, cpu_engine_); in Compute()
162 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklRequantizePerChannelOp
|
| D | mkl_relu_op.cc | 62 : cpu_engine_(engine::cpu, 0) { in MklEltwiseFwdPrimitive()
141 new memory::primitive_desc(*context_.src_md, cpu_engine_)); in Setup()
148 *context_.fwd_desc, cpu_engine_)); in Setup()
163 engine cpu_engine_; member in tensorflow::MklEltwiseFwdPrimitive
246 : cpu_engine_(engine::cpu, 0) { in MklEltwiseBwdPrimitive()
346 new memory::primitive_desc(*context_.src_md, cpu_engine_)); in Setup()
348 new memory::primitive_desc(*context_.diff_dst_md, cpu_engine_)); in Setup()
355 *context_.fwd_desc, cpu_engine_)); in Setup()
360 *context_.bwd_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
377 engine cpu_engine_; member in tensorflow::MklEltwiseBwdPrimitive
|
| D | mkl_avgpooling_op.cc | 452 MklDnnData<T> dnn_data_input(&cpu_engine_); in Compute()
524 cpu_engine_); in Compute()
565 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklAvgPoolingOp
587 MklDnnData<T> grad_dnn_data(&cpu_engine_); in Compute()
645 cpu_engine_); in Compute()
670 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklAvgPoolingGradOp
|
| D | mkl_conv_grad_input_ops.cc | 83 : cpu_engine_(engine::cpu, 0) { in MklConvBwdInputPrimitive()
200 *context_.fwd_desc, cpu_engine_)); in Setup()
204 *context_.bwd_input_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
235 engine cpu_engine_; member in tensorflow::MklConvBwdInputPrimitive
|
| D | mkl_conv_ops.cc | 105 : cpu_engine_(engine::cpu, 0) { in MklConvFwdPrimitive()
244 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
276 new ConvFwdPd(*context_.fwd_desc, post_ops_attr, cpu_engine_)); in Setup()
278 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
300 cpu_engine_}, in Setup()
316 engine cpu_engine_; member in tensorflow::MklConvFwdPrimitive
937 MklDnnData<Tinput> src(&cpu_engine_); in Compute()
938 MklDnnData<Tfilter> filter(&cpu_engine_); in Compute()
1261 dst_pd = memory::primitive_desc(dst_md, cpu_engine_); in AllocateOutputTensor()
1279 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklConvOp
[all …]
|
| D | mkl_slice_op.cc | 217 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklSlicePrimitive
223 new memory({sliceParams.from->get_primitive_desc().desc(), cpu_engine_}, in Setup()
226 {sliceParams.to->get_primitive_desc().desc(), cpu_engine_}, DummyData)); in Setup()
|
| D | mkl_pooling_ops_common.cc | 68 new pooling_forward::primitive_desc(*context_.fwd_desc, cpu_engine_)); in Setup()
76 {{{fwdParams.src_dims}, MklDnnType<T>(), context_.src_fmt}, cpu_engine_}, in Setup()
|
| D | mkl_pooling_ops_common.h | 73 : cpu_engine_(engine::cpu, 0) { in MklPoolingFwdPrimitive()
149 engine cpu_engine_; variable
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1664 const engine* cpu_engine_; variable
1672 cpu_engine_(e) {} in MklDnnData()
1678 cpu_engine_ = nullptr; // We don't own this. in ~MklDnnData()
1751 auto pd = memory::primitive_desc(md, *cpu_engine_);
1767 CHECK_NOTNULL(cpu_engine_);
2281 engine cpu_engine_ = engine(engine::cpu, 0); variable
2285 {from->get_primitive_desc().desc(), cpu_engine_}, DummyData)); in Setup()
2287 new memory({to->get_primitive_desc().desc(), cpu_engine_}, DummyData)); in Setup()
|