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