| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_aggregate_ops.cc | 172 src1.SetUsrMem(md1, &src1_tensor); in Compute()
173 src2.SetUsrMem(md2, &src2_tensor); in Compute()
184 dst.SetUsrMem(common_md); in Compute()
189 dst.SetUsrMem(common_md); in Compute()
|
| D | mkl_lrn_op.cc | 133 src_dnn_data.SetUsrMem(src_md, &src_tensor); in Compute()
137 dst_dnn_data.SetUsrMem(src_md); in Compute()
285 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor); in AllocateWorkspaceTensor()
388 orig_input_dnn_data.SetUsrMem(orig_input_md, &orig_input_tensor); in Compute()
392 output_dnn_data.SetUsrMem(orig_input_md); in Compute()
478 input_grad_dnn_data->SetUsrMem(input_grad_md, &input_grad_tensor); in ConfigureInputGradient()
516 workspace_dnn_data->SetUsrMem(workspace_pd, &workspace_tensor); in ConfigureWorkspace()
|
| D | mkl_softmax_op.cc | 127 src.SetUsrMem(src_md, &src_tensor); in Compute()
160 dst.SetUsrMem(src_md, output_tensor); in Compute()
|
| D | mkl_maxpooling_op.cc | 591 dnn_data_output.SetUsrMem(output_dims_mkl_order, in Compute()
598 dnn_data_input.SetUsrMem(input_md, &input_tensor); in Compute()
671 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor); in AllocateWorkspaceTensor()
757 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor); in Compute()
776 workspace_dnn_data.SetUsrMem(ws_md, &workspace_tensor); in Compute()
809 workspace_dnn_data->SetUsrMem(workspace_pd, &workspace_tensor); in ConfigureWorkspace()
|
| D | mkl_transpose_op.cc | 138 in.SetUsrMem(in_dims, in_strides, &in_tensor); in MKLTransposeND()
141 out.SetUsrMem(in_dims, out_strides, out_tensor); in MKLTransposeND()
|
| D | mkl_input_conversion_op.cc | 142 input.SetUsrMem(input0_md, &input_tensor_0); in Compute()
241 tf_input.SetUsrMem(input_tf_md, tf_tensor); in Compute()
|
| D | mkl_concat_op.cc | 306 srcs[k].SetUsrMem(src_md, &input_tensors[k]); in Compute()
317 srcs[k].SetUsrMem(src_md, &input_tensors[k]); in Compute()
341 srcs[k].SetUsrMem(src_md, &input_tensors[k]); in Compute()
415 dst.SetUsrMem(dst_md, dst_tensor); in Compute()
|
| D | mkl_slice_op.cc | 401 src.SetUsrMem(input_md, &input_tensor); in ComputeMklSlice()
411 src.SetUsrMem(input_md, &input_tensor); in ComputeMklSlice()
423 output.SetUsrMem(output_md, output_tensor); in ComputeMklSlice()
|
| D | mkl_tfconv_op.h | 95 input.SetUsrMem(input_mkl_md, &input_tensor); in ConvertMklToTf()
|
| D | mkl_conv_grad_filter_ops.cc | 540 src.SetUsrMem(fwd_src_md, &src_tensor); in Compute()
550 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
565 diff_filter.SetUsrMem(bwd_output_dims, GetOutputFormat(tf_fmt), in Compute()
|
| D | mkl_reshape_op.cc | 148 dnn_data_input.SetUsrMem(input_mkl_md, &input_tensor); in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 442 filter.SetUsrMem(fwd_filter_md, &filter_tensor); in Compute()
452 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
|
| D | mkl_avgpooling_op.cc | 521 dnn_data_input.SetUsrMem(input_md, &input_tensor); in Compute()
646 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 647 src.SetUsrMem(src_md, &src_tensor); in Compute()
964 src.SetUsrMem(src_md, &src_tensor); in Compute()
977 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
|
| D | mkl_relu_op.cc | 493 src.SetUsrMem(src_md, &src_tensor); in Compute()
650 src.SetUsrMem(src_md, &src_tensor); in Compute()
659 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
|
| D | mkl_conv_ops.cc | 1033 src.SetUsrMem(src_md, &src_tensor); in Compute()
1048 filter.SetUsrMem(filter_md, &filter_tensor); in Compute()
1102 src.SetUsrMem(src_md, &src_tensor); in Compute()
1128 filter.SetUsrMem(filter_md, &filter_tensor); in Compute()
1418 filter.SetUsrMem(filter_md, &filter_tensor); in CacheFilter()
|
| D | mkl_pooling_ops_common.h | 623 dnn_data_input->SetUsrMem(input_md, &input_tensor); in ConfigureInput()
732 input_gradient_dnn_data->SetUsrMem(original_input_grad_md, in ConfigureInputGradient()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 66 a.SetUsrMem(dim1, strides1); in TEST()
78 b.SetUsrMem(dim2, strides2); in TEST()
|
| D | mkl_util.h | 852 input.SetUsrMem(input_mkl_md, &mkl_tensor); in ConvertMklToTF()
1702 inline void SetUsrMem(const memory::dims& dim, memory::format fm,
1705 SetUsrMem(md, data_buffer);
1708 inline void SetUsrMem(const memory::dims& dim, memory::format fm, in SetUsrMem() function
1711 SetUsrMem(dim, fm, GetTensorBuffer(tensor)); in SetUsrMem()
1733 inline void SetUsrMem(const memory::dims& dim, const memory::dims& strides,
1737 SetUsrMem(blocked_md, data_buffer);
1740 inline void SetUsrMem(const memory::dims& dim, const memory::dims& strides, in SetUsrMem() function
1743 SetUsrMem(dim, strides, GetTensorBuffer(tensor)); in SetUsrMem()
1750 inline void SetUsrMem(const memory::desc& md, void* data_buffer = nullptr) {
[all …]
|