| /external/tensorflow/tensorflow/core/util/ |
| mkl_util_test.cc | 29 MklDnnData<float> a(&cpu_engine);
63 MklDnnData<float> a(&cpu_engine);
75 MklDnnData<float> b(&cpu_engine);
|
| mkl_util.h | [all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_pooling_ops_common.h | 198 MklDnnData<T>* dnn_data_input) {
238 const MklDnnData<T>* src, MklDnnData<T>* dst,
239 MklDnnData<uint8>* wksp = nullptr) {
304 MklDnnData<T>* input_gradient_diff_dst, MklDnnData<T>* output_diff_src,
306 const MklDnnData<uint8>* workspace = nullptr) {
366 MklDnnData<T>* input_gradient_dnn_data,
|
| mkl_softmax_op.cc | 71 MklDnnData<T> src(&cpu_engine);
72 MklDnnData<T> dst(&cpu_engine);
|
| mkl_conv_grad_filter_ops.cc | 493 MklDnnData<T>* input, MklDnnData<T>* filter,
494 MklDnnData<T>* outbackprop, MklDnnData<T>* output,
507 MklDnnData<T>* bias_grad = nullptr;
511 bias_grad = new MklDnnData<T>(&cpu_engine);
606 MklDnnData<T>* input, MklDnnData<T>* obp, MklDnnData<T>* output,
607 MklDnnData<T>* bias_grad = nullptr)
[all...] |
| mkl_conv_ops.h | 357 MklDnnData<T> input(&cpu_engine);
358 MklDnnData<T> filter(&cpu_engine);
359 MklDnnData<T> outbackprop(&cpu_engine);
360 MklDnnData<T> output(&cpu_engine);
541 MklDnnData<T>* input, MklDnnData<T>* filter, MklDnnData<T>* outbackprop,
542 MklDnnData<T>* output, Tensor** output_tensor,
|
| mkl_conv_grad_input_ops.cc | 420 MklDnnData<T>* input, MklDnnData<T>* filter,
421 MklDnnData<T>* outbackprop, MklDnnData<T>* output,
482 MklDnnData<T>* filter, MklDnnData<T>* obp, MklDnnData<T>* output) {
|
| mkl_lrn_op.cc | 790 MklDnnData<T> src_dnn_data(&cpu_engine);
791 MklDnnData<T> dst_dnn_data(&cpu_engine);
792 MklDnnData<uint8> workspace_dnn_data(&cpu_engine);
[all...] |
| mkl_relu_op.cc | 398 MklDnnData<T> src(&cpu_engine);
399 MklDnnData<T> dst(&cpu_engine);
409 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides);
475 MklDnnData<T> src(&cpu_engine);
476 MklDnnData<T> diff_dst(&cpu_engine);
477 MklDnnData<T> diff_src(&cpu_engine);
515 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides);
614 MklDnnData<T>* src, MklDnnData<T>* diff_src,
615 MklDnnData<T>* diff_dst)
[all...] |
| mkl_conv_ops.cc | 527 MklDnnData<T> src(&cpu_engine);
528 MklDnnData<T> filter(&cpu_engine);
529 MklDnnData<T> output(&cpu_engine);
599 MklDnnData<T> bias(&cpu_engine);
720 MklDnnData<T>* src, MklDnnData<T>* filter, MklDnnData<T>* bias,
721 MklDnnData<T>* output, Tensor* filter_out_tensor) {
[all...] |
| mkl_maxpooling_op.cc | 520 MklDnnData<T> dnn_data_input(&cpu_engine);
521 MklDnnData<T> dnn_data_output(&cpu_engine);
522 MklDnnData<uint8> dnn_data_wksp(&cpu_engine);
589 MklDnnData<uint8>* dnn_data_wksp) {
643 MklDnnData<T> grad_dnn_data(&cpu_engine);
644 MklDnnData<uint8> workspace_dnn_data(&cpu_engine);
645 MklDnnData<T> output_dnn_data(&cpu_engine);
735 MklDnnData<uint8>* workspace_dnn_data) {
[all...] |
| mkl_aggregate_ops.cc | 359 MklDnnData<T> src1(&cpu_engine);
360 MklDnnData<T> src2(&cpu_engine);
361 MklDnnData<T> dst(&cpu_engine);
383 md1 = MklDnnData<T>::CreateBlockedMemDesc(dims, strides);
|
| mkl_tfconv_op.h | 90 MklDnnData<T> input(&cpu_engine);
|
| mkl_reshape_op.cc | 228 MklDnnData<T> dnn_data_input(&cpu_engine);
279 auto output_tf_md = MklDnnData<T>::CreateBlockedMemDesc(
|
| mkl_input_conversion_op.cc | 331 // Create MklDnnData object for input0 tesnsor
333 MklDnnData<T> input(&cpu_engine);
438 // Create MklDnnData object for input tensor. Input tensor is in
441 MklDnnData<T> tf_input(&cpu_engine);
|
| mkl_avgpooling_op.cc | 453 MklDnnData<T> dnn_data_input(&cpu_engine);
454 MklDnnData<T> dnn_data_output(&cpu_engine);
570 MklDnnData<T> input_gradient_diff_dst(&cpu_engine);
571 MklDnnData<T> output_diff_src(&cpu_engine);
|
| mkl_fused_batch_norm_op.cc | [all...] |
| mkl_concat_op.cc | 685 std::vector<MklDnnData<T>> srcs(N, MklDnnData<T>(&cpu_engine));
710 MklDnnData<T> dst(&cpu_engine);
[all...] |
