| /external/tensorflow/tensorflow/core/util/ |
| mkl_util_test.cc | 28 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/ |
| mkl_softmax_op.cc | 51 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);
|
| mkl_maxpooling_op.cc | 512 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...] |
| mkl_tfconv_op.h | 89 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);
|
| mkl_avgpooling_op.cc | 445 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...] |
| mkl_fused_batch_norm_op.cc | 705 auto cpu_engine = engine(engine::cpu, 0); variable
789 MklDnnData<T> src(&cpu_engine); variable
790 MklDnnData<T> dst(&cpu_engine); variable
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...] |
| mkl_aggregate_ops.cc | 304 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));
|
| mkl_input_conversion_op.cc | 332 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),
|
| mkl_conv_ops.h | 353 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,
|
| mkl_lrn_op.cc | 764 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);
[all...] |
| mkl_relu_op.cc | 384 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_pd)
[all...] |
| mkl_conv_ops.cc | 514 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);
[all...] |
| mkl_reshape_op.cc | 227 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);
|
| mkl_conv_grad_input_ops.cc | 418 void CreatePrimitive(OpKernelContext* context, const engine& cpu_engine,
440 bwd_desc, cpu_engine, conv_fwd_pd);
|
| mkl_conv_grad_filter_ops.cc | 491 void CreatePrimitive(OpKernelContext* context, const engine& cpu_engine,
511 bias_grad = new MklDnnData<T>(&cpu_engine);
535 bwd_desc, cpu_engine, conv_fwd_pd);
|
| mkl_concat_op.cc | 582 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);
[all...] |
