| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_lrn_op.cc | 120 MklDnnData<T> src_dnn_data(&cpu_engine); in Compute()
121 MklDnnData<T> dst_dnn_data(&cpu_engine); in Compute()
122 MklDnnData<uint8> workspace_dnn_data(&cpu_engine); in Compute()
175 MklDnnData<T>* src_dnn_data, in PrepareAndExecuteNet()
176 MklDnnData<T>* dst_dnn_data, in PrepareAndExecuteNet()
177 MklDnnData<uint8>* wksp_dnn_data = nullptr) { in PrepareAndExecuteNet()
270 MklDnnData<uint8>* dnn_data_wksp) { in AllocateWorkspaceTensor()
344 MklDnnData<T> input_grad_dnn_data(&cpu_engine); in Compute()
345 MklDnnData<T> orig_input_dnn_data(&cpu_engine); in Compute()
346 MklDnnData<T> orig_output_dnn_data(&cpu_engine); in Compute()
[all …]
|
| D | mkl_aggregate_ops.cc | 105 MklDnnData<T> src1(&cpu_engine); in Compute()
106 MklDnnData<T> src2(&cpu_engine); in Compute()
107 MklDnnData<T> dst(&cpu_engine); in Compute()
129 md1 = MklDnnData<T>::CreateBlockedMemDesc(dims, strides); in Compute()
|
| D | mkl_softmax_op.cc | 114 MklDnnData<T> src(&cpu_engine); in Compute()
115 MklDnnData<T> dst(&cpu_engine); in Compute()
|
| D | mkl_maxpooling_op.cc | 521 MklDnnData<T> dnn_data_input(&cpu_engine); in Compute()
522 MklDnnData<T> dnn_data_output(&cpu_engine); in Compute()
631 MklDnnData<uint8> dnn_data_wksp(&cpu_engine); in Compute()
657 MklDnnData<uint8>* dnn_data_wksp) { in AllocateWorkspaceTensor()
700 MklDnnData<T> grad_dnn_data(&cpu_engine); in Compute()
701 MklDnnData<uint8> workspace_dnn_data(&cpu_engine); in Compute()
806 MklDnnData<uint8>* workspace_dnn_data) { in ConfigureWorkspace()
|
| D | mkl_slice_op.cc | 362 MklDnnData<T> src(&cpu_engine); in ComputeMklSlice()
363 MklDnnData<T> output(&cpu_engine); in ComputeMklSlice()
410 MklDnnData<T>::CreateBlockedMemDesc(input_dims, input_strides); in ComputeMklSlice()
417 MklDnnData<T>::CreateBlockedMemDesc(size_dims, output_strides); in ComputeMklSlice()
|
| D | mkl_reshape_op.cc | 130 MklDnnData<T> dnn_data_input(&cpu_engine); in Compute()
180 auto output_tf_md = MklDnnData<T>::CreateBlockedMemDesc( in Compute()
|
| D | mkl_transpose_op.cc | 128 MklDnnData<T> in(&cpu_engine); in MKLTransposeND()
129 MklDnnData<T> out(&cpu_engine); in MKLTransposeND()
|
| D | mkl_input_conversion_op.cc | 141 MklDnnData<T> input(&cpu_engine); in Compute()
239 MklDnnData<T> tf_input(&cpu_engine); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 595 MklDnnData<T> src(&cpu_engine); in Compute()
596 MklDnnData<T> weights(&cpu_engine); in Compute()
919 MklDnnData<T> src(&cpu_engine); in Compute()
920 MklDnnData<T> diff_dst(&cpu_engine); in Compute()
921 MklDnnData<T> weights(&cpu_engine); in Compute()
922 MklDnnData<T> diff_weights(&cpu_engine); in Compute()
|
| D | mkl_concat_op.cc | 290 std::vector<MklDnnData<T>> srcs(N, MklDnnData<T>(&cpu_engine)); in Compute()
348 MklDnnData<T> dst(&cpu_engine); in Compute()
|
| D | mkl_tfconv_op.h | 86 MklDnnData<T> input(&cpu_engine); in ConvertMklToTf()
|
| D | mkl_relu_op.cc | 470 MklDnnData<T> src(&cpu_engine); in Compute()
480 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides); in Compute()
564 MklDnnData<T> src(&cpu_engine); in Compute()
565 MklDnnData<T> diff_dst(&cpu_engine); in Compute()
592 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 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()
|
| D | mkl_conv_ops.cc | 937 MklDnnData<Tinput> src(&cpu_engine_); in Compute()
938 MklDnnData<Tfilter> filter(&cpu_engine_); in Compute()
1357 MklDnnData<Tinput>* src, in PrepareAndExecuteNet()
1358 MklDnnData<Tfilter>* filter, in PrepareAndExecuteNet()
1359 MklDnnData<Tbias>* bias, in PrepareAndExecuteNet()
1360 MklDnnData<Toutput>* output, in PrepareAndExecuteNet()
1406 MklDnnData<Tfilter>& filter, const memory::desc& filter_md) in CacheFilter()
|
| D | mkl_conv_grad_input_ops.cc | 309 MklDnnData<T> filter(&cpu_engine); in Compute()
310 MklDnnData<T> diff_dst(&cpu_engine); in Compute()
|
| D | mkl_avgpooling_op.cc | 452 MklDnnData<T> dnn_data_input(&cpu_engine_); in Compute()
587 MklDnnData<T> grad_dnn_data(&cpu_engine_); in Compute()
|
| D | mkl_pooling_ops_common.h | 608 MklDnnData<T>* dnn_data_input) { in ConfigureInput()
717 MklDnnData<T>* input_gradient_dnn_data, in ConfigureInputGradient()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 29 MklDnnData<float> a(&cpu_engine); in TEST()
63 MklDnnData<float> a(&cpu_engine); in TEST()
75 MklDnnData<float> b(&cpu_engine); in TEST()
|
| D | mkl_util.h | 829 class MklDnnData; variable
846 MklDnnData<T> input(&cpu_engine); in ConvertMklToTF()
1649 class MklDnnData {
1667 explicit MklDnnData(const engine* e) in MklDnnData() function
1674 ~MklDnnData() { in ~MklDnnData()
1736 auto blocked_md = MklDnnData<T>::CreateBlockedMemDesc(dim, strides);
|