Searched refs:cpu_engine (Results 1 - 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/util/
H A Dmkl_util_test.cc28 auto cpu_engine = engine(engine::cpu, 0); local
29 MklDnnData<float> a(&cpu_engine);
60 auto cpu_engine = engine(engine::cpu, 0); local
63 MklDnnData<float> a(&cpu_engine);
75 MklDnnData<float> b(&cpu_engine);
/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_softmax_op.cc51 auto cpu_engine = engine(engine::cpu, 0); variable
71 MklDnnData<T> src(&cpu_engine); variable
72 MklDnnData<T> dst(&cpu_engine); variable
97 softmax_forward::primitive_desc(softmax_fwd_desc, cpu_engine);
H A Dmkl_input_conversion_op.cc332 auto cpu_engine = engine(engine::cpu, 0); variable
333 MklDnnData<T> input(&cpu_engine); variable
339 memory::primitive_desc(input1_md, cpu_engine),
440 auto cpu_engine = engine(engine::cpu, 0); variable
441 MklDnnData<T> tf_input(&cpu_engine); variable
448 memory::primitive_desc(output_mkl_md, cpu_engine),
H A Dmkl_conv_ops.h353 auto cpu_engine = engine(engine::cpu, 0); variable
357 MklDnnData<T> input(&cpu_engine); variable
358 MklDnnData<T> filter(&cpu_engine); variable
359 MklDnnData<T> outbackprop(&cpu_engine); variable
360 MklDnnData<T> output(&cpu_engine); variable
444 auto fwd_pd = convolution_forward::primitive_desc(fwd_desc, cpu_engine);
487 CreatePrimitive(context, cpu_engine, fwd_pd, &input, &filter,
539 OpKernelContext* context, const engine& cpu_engine,
H A Dmkl_reshape_op.cc227 auto cpu_engine = engine(engine::cpu, 0); variable
228 MklDnnData<T> dnn_data_input(&cpu_engine); variable
250 memory::primitive_desc(output_tf_md, cpu_engine);
282 memory::primitive_desc(output_tf_md, cpu_engine);
H A Dmkl_maxpooling_op.cc512 auto cpu_engine = engine(engine::cpu, 0); variable
520 MklDnnData<T> dnn_data_input(&cpu_engine);
521 MklDnnData<T> dnn_data_output(&cpu_engine); variable
522 MklDnnData<uint8> dnn_data_wksp(&cpu_engine); variable
562 pooling_forward::primitive_desc(pool_desc, cpu_engine);
621 auto cpu_engine = engine(engine::cpu, 0); variable
643 MklDnnData<T> grad_dnn_data(&cpu_engine);
644 MklDnnData<uint8> workspace_dnn_data(&cpu_engine); variable
645 MklDnnData<T> output_dnn_data(&cpu_engine); variable
675 pooling_forward::primitive_desc(pool_fwd_desc, cpu_engine);
[all...]
H A Dmkl_tfconv_op.h89 auto cpu_engine = engine(engine::cpu, 0); local
90 MklDnnData<T> input(&cpu_engine);
97 auto output_tf_pd = memory::primitive_desc(output_tf_md, cpu_engine);
H A Dmkl_avgpooling_op.cc445 auto cpu_engine = engine(engine::cpu, 0); variable
453 MklDnnData<T> dnn_data_input(&cpu_engine);
454 MklDnnData<T> dnn_data_output(&cpu_engine); variable
518 pooling_forward::primitive_desc(pool_desc, cpu_engine);
550 auto cpu_engine = engine(engine::cpu, 0); variable
570 MklDnnData<T> input_gradient_diff_dst(&cpu_engine);
571 MklDnnData<T> output_diff_src(&cpu_engine); variable
612 pooling_forward::primitive_desc(pool_fwd_desc, cpu_engine);
625 pool_bkwd_desc, cpu_engine, pool_fwd_prim_desc);
634 memory::primitive_desc(target_diff_dst_md, cpu_engine));
[all...]
H A Dmkl_relu_op.cc384 auto cpu_engine = engine(engine::cpu, 0); variable
398 MklDnnData<T> src(&cpu_engine); variable
399 MklDnnData<T> dst(&cpu_engine); variable
420 new relu_forward::primitive_desc(relu_fwd_desc, cpu_engine));
474 auto cpu_engine = engine(engine::cpu, 0); local
475 MklDnnData<T> src(&cpu_engine);
476 MklDnnData<T> diff_dst(&cpu_engine);
477 MklDnnData<T> diff_src(&cpu_engine);
573 new relu_forward::primitive_desc(relu_fwd_desc, cpu_engine));
577 relu_bwd_desc, cpu_engine, *relu_fwd_p
[all...]
H A Dmkl_fused_batch_norm_op.cc705 auto cpu_engine = engine(engine::cpu, 0); variable
789 MklDnnData<T> src(&cpu_engine); variable
790 MklDnnData<T> dst(&cpu_engine); variable
823 auto weights_pd = memory::primitive_desc(weights_desc, cpu_engine);
839 auto mean_pd = memory::primitive_desc(mean_desc, cpu_engine);
850 auto variance_pd = memory::primitive_desc(variance_desc, cpu_engine);
865 bnrm_fwd_desc, cpu_engine);
1048 auto cpu_engine = engine(engine::cpu, 0); variable
1128 MklDnnData<T> src(&cpu_engine); variable
1129 MklDnnData<T> mean(&cpu_engine); variable
1130 MklDnnData<T> variance(&cpu_engine); variable
1131 MklDnnData<T> diff_dst(&cpu_engine); variable
1132 MklDnnData<T> diff_src(&cpu_engine); variable
[all...]
H A Dmkl_aggregate_ops.cc304 auto cpu_engine = engine(engine::cpu, 0); variable
359 MklDnnData<T> src1(&cpu_engine); variable
360 MklDnnData<T> src2(&cpu_engine); variable
361 MklDnnData<T> dst(&cpu_engine); variable
436 srcs_pd.push_back(memory::primitive_desc(common_md, cpu_engine));
438 srcs_pd.push_back(memory::primitive_desc(common_md, cpu_engine));
H A Dmkl_lrn_op.cc764 auto cpu_engine = engine(engine::cpu, 0);
790 MklDnnData<T> src_dnn_data(&cpu_engine);
791 MklDnnData<T> dst_dnn_data(&cpu_engine);
792 MklDnnData<uint8> workspace_dnn_data(&cpu_engine);
816 auto lrn_prim_desc = lrn_forward::primitive_desc(lrn_desc, cpu_engine);
1013 auto cpu_engine = engine(engine::cpu, 0);
1014 MklDnnData<T> input_grad_dnn_data(&cpu_engine);
1015 MklDnnData<T> orig_input_dnn_data(&cpu_engine);
1016 MklDnnData<T> orig_output_dnn_data(&cpu_engine);
1017 MklDnnData<T> output_dnn_data(&cpu_engine);
[all...]
H A Dmkl_conv_ops.cc514 auto cpu_engine = engine(engine::cpu, 0); variable
527 MklDnnData<T> src(&cpu_engine); variable
528 MklDnnData<T> filter(&cpu_engine); variable
529 MklDnnData<T> output(&cpu_engine); variable
599 MklDnnData<T> bias(&cpu_engine); variable
613 convolution_forward::primitive_desc(conv_desc, cpu_engine);
634 convolution_forward::primitive_desc(conv_desc, cpu_engine);
H A Dmkl_conv_grad_filter_ops.cc491 void CreatePrimitive(OpKernelContext* context, const engine& cpu_engine, argument
511 bias_grad = new MklDnnData<T>(&cpu_engine);
535 bwd_desc, cpu_engine, conv_fwd_pd);
H A Dmkl_conv_grad_input_ops.cc418 void CreatePrimitive(OpKernelContext* context, const engine& cpu_engine, argument
440 bwd_desc, cpu_engine, conv_fwd_pd);
H A Dmkl_concat_op.cc582 auto cpu_engine = engine(engine::cpu, 0); local
685 std::vector<MklDnnData<T>> srcs(N, MklDnnData<T>(&cpu_engine));
710 MklDnnData<T> dst(&cpu_engine);

Completed in 218 milliseconds