| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 28 MklDnnData<float> a(&cpu_engine); in TEST()
62 MklDnnData<float> a(&cpu_engine); in TEST()
73 MklDnnData<float> b(&cpu_engine); in TEST()
|
| D | mkl_util.h | 606 class MklDnnData; variable
640 MklDnnData<T> input(&cpu_engine); in ConvertMklToTF()
1249 class MklDnnData {
1267 explicit MklDnnData(const engine* e) in MklDnnData() function
1275 ~MklDnnData() { in ~MklDnnData()
1339 auto blocked_md = MklDnnData<T>::CreateBlockedMemDesc(dim, strides);
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 124 MklDnnData<T> src_dnn_data(&cpu_engine_); in Compute()
125 MklDnnData<T> dst_dnn_data(&cpu_engine_); in Compute()
126 MklDnnData<uint8> workspace_dnn_data(&cpu_engine_); in Compute()
264 MklDnnData<uint8>* dnn_data_wksp) { in AllocateWorkspaceTensor()
340 MklDnnData<T> input_grad_dnn_data(&cpu_engine_); in Compute()
341 MklDnnData<T> orig_input_dnn_data(&cpu_engine_); in Compute()
342 MklDnnData<T> orig_output_dnn_data(&cpu_engine_); in Compute()
343 MklDnnData<T> output_dnn_data(&cpu_engine_); in Compute()
422 MklDnnData<uint8> workspace_dnn_data(&cpu_engine_); in Compute()
478 MklDnnData<T>* input_grad_dnn_data) { in ConfigureInputGradient()
[all …]
|
| D | mkl_maxpooling_op.cc | 64 MklDnnData<T> dnn_data_input(&cpu_engine_); in Compute()
65 MklDnnData<T> dnn_data_output(&cpu_engine_); in Compute()
180 MklDnnData<uint8> dnn_data_wksp(&cpu_engine_); in Compute()
205 MklDnnData<uint8>* dnn_data_wksp) { in AllocateWorkspaceTensor()
250 MklDnnData<T> grad_dnn_data(&cpu_engine_); in Compute()
251 MklDnnData<uint8> workspace_dnn_data(&cpu_engine_); in Compute()
|
| D | mkl_aggregate_ops.cc | 161 MklDnnData<T> dst(&cpu_engine); in Compute()
162 MklDnnData<T> src(&cpu_engine); in Compute()
208 md = MklDnnData<T>::CreateBlockedMemDesc(dims, strides); in Compute()
|
| D | mkl_slice_op.cc | 370 MklDnnData<T> src(&cpu_engine); in ComputeMklSlice()
371 MklDnnData<T> output(&cpu_engine); in ComputeMklSlice()
428 MklDnnData<T>::CreateBlockedMemDesc(input_dims, input_strides); in ComputeMklSlice()
435 MklDnnData<T>::CreateBlockedMemDesc(input_dims, input_strides); in ComputeMklSlice()
441 MklDnnData<T>::CreateBlockedMemDesc(size_dims, output_strides); in ComputeMklSlice()
|
| D | mkl_dequantize_op.cc | 74 MklDnnData<T> src(&cpu_engine); in Compute()
75 MklDnnData<float> dst(&cpu_engine); in Compute()
|
| D | mkl_input_conversion_op.cc | 140 MklDnnData<T> input(&cpu_engine); in Compute()
241 MklDnnData<T> tf_input(&cpu_engine); in Compute()
|
| D | mkl_transpose_op.cc | 129 MklDnnData<T> in(&cpu_engine); in MKLTransposeND()
130 MklDnnData<T> out(&cpu_engine); in MKLTransposeND()
|
| D | mkl_fused_batch_norm_op.cc | 747 MklDnnData<T> src(&cpu_engine_); in Compute()
748 MklDnnData<U> weights(&cpu_engine_); in Compute()
749 MklDnnData<U> wksp(&cpu_engine_); in Compute()
1156 MklDnnData<T> src(&cpu_engine_); in Compute()
1157 MklDnnData<T> diff_dst(&cpu_engine_); in Compute()
1158 MklDnnData<U> weights(&cpu_engine_); in Compute()
1159 MklDnnData<U> diff_weights(&cpu_engine_); in Compute()
1181 MklDnnData<T> reorder_src(&cpu_engine_); in Compute()
1182 MklDnnData<T> reorder_diff_dst(&cpu_engine_); in Compute()
|
| D | mkl_tfconv_op.h | 89 MklDnnData<T> input(&cpu_engine); in ConvertMklToTf()
|
| D | mkl_matmul_op_fused.cc | 209 MklDnnData<T> src_mkl(&(this->cpu_engine_)); in Compute()
210 MklDnnData<T> weight_mkl(&(this->cpu_engine_)); in Compute()
|
| D | mkl_avgpooling_op.cc | 60 MklDnnData<T> dnn_data_input(&cpu_engine_); in Compute()
192 MklDnnData<T> grad_dnn_data(&cpu_engine_); in Compute()
|
| D | mkl_concat_op.cc | 591 std::vector<MklDnnData<T>> srcs(N, MklDnnData<T>(&cpu_engine)); in Compute()
663 MklDnnData<T> dst(&cpu_engine); in Compute()
679 dst_md = MklDnnData<T>::CreateBlockedMemDesc( in Compute()
|
| D | mkl_relu_op.cc | 462 MklDnnData<T> src(&cpu_engine); in Compute()
473 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides); in Compute()
587 MklDnnData<T> src(&cpu_engine); in Compute()
588 MklDnnData<T> diff_dst(&cpu_engine); in Compute()
628 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides); in Compute()
|
| D | mkl_reshape_op.cc | 141 MklDnnData<T> dnn_data_input(&cpu_engine); in Compute()
|
| D | mkl_quantize_op.cc | 368 MklDnnData<float> src(&cpu_engine); in Compute()
369 MklDnnData<T> dst(&cpu_engine); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 539 MklDnnData<T> src(&cpu_engine_); in Compute()
551 MklDnnData<T> diff_dst(&cpu_engine_); in Compute()
569 MklDnnData<T> diff_filter(&cpu_engine_); in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 461 MklDnnData<T> filter(&cpu_engine_); in Compute()
473 MklDnnData<T> diff_dst(&cpu_engine_); in Compute()
|
| D | mkl_qmatmul_op.cc | 174 MklDnnData<Tinput> src(&(this->cpu_engine_)); in Compute()
175 MklDnnData<Tweight> weight(&(this->cpu_engine_)); in Compute()
|
| D | mkl_conv_ops.cc | 511 MklDnnData<Tinput> src(&cpu_engine_); in Compute()
512 MklDnnData<Tfilter> filter(&cpu_engine_); in Compute()
1040 MklDnnData<Tinput>* src, in PrepareAndExecuteNet()
1041 MklDnnData<Tfilter>* filter, in PrepareAndExecuteNet()
1042 MklDnnData<Tbias>* bias, in PrepareAndExecuteNet()
1043 MklDnnData<Toutput>* output, in PrepareAndExecuteNet()
1093 MklDnnData<Tfilter>& filter, const memory::desc& filter_md, in CacheFilter()
|
| D | mkl_pooling_ops_common.h | 610 MklDnnData<T>* dnn_data_input) { in ConfigureInput()
|
| D | mkl_matmul_ops_common.h | 428 MklDnnData<Tweight>& weight, const memory::desc& weight_md) in CacheWeight()
|