| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 65 a.SetUsrMem(dim1, strides1); in TEST()
76 b.SetUsrMem(dim2, strides2); in TEST()
|
| D | mkl_util.h | 645 input.SetUsrMem(input_mkl_md, &input_mkl_tensor); in ConvertMklToTF()
1302 inline void SetUsrMem(const memory::dims& dim, memory::format_tag fm,
1305 SetUsrMem(md, data_buffer);
1308 inline void SetUsrMem(const memory::dims& dim, memory::format_tag fm, in SetUsrMem() function
1311 SetUsrMem(dim, fm, GetTensorBuffer(tensor)); in SetUsrMem()
1336 inline void SetUsrMem(const memory::dims& dim, const memory::dims& strides,
1340 SetUsrMem(blocked_md, data_buffer);
1343 inline void SetUsrMem(const memory::dims& dim, const memory::dims& strides, in SetUsrMem() function
1346 SetUsrMem(dim, strides, GetTensorBuffer(tensor)); in SetUsrMem()
1350 inline void SetUsrMem(const memory::desc& md, const Tensor* tensor) { in SetUsrMem() function
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 137 src_dnn_data.SetUsrMem(src_md, &src_tensor); in Compute()
142 dst_dnn_data.SetUsrMem(src_md); in Compute()
278 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor); in AllocateWorkspaceTensor()
384 orig_input_dnn_data.SetUsrMem(orig_input_md, &orig_input_tensor); in Compute()
390 output_dnn_data.SetUsrMem(orig_input_md); in Compute()
485 input_grad_dnn_data->SetUsrMem(input_grad_md, &input_grad_tensor); in ConfigureInputGradient()
496 workspace_dnn_data->SetUsrMem(workspace_pd, &workspace_tensor); in ConfigureWorkspace()
|
| D | mkl_dequantize_op.cc | 89 src.SetUsrMem(src_md, &src_tensor); in Compute()
124 dst.SetUsrMem(dst_md, output_tensor); in Compute()
|
| D | mkl_input_conversion_op.cc | 141 input.SetUsrMem(input0_md, &input_tensor_0); in Compute()
243 tf_input.SetUsrMem(input_tf_md, tf_tensor); in Compute()
|
| D | mkl_qmatmul_op.cc | 213 src.SetUsrMem(src_md, &src_tensor); in Compute()
221 weight.SetUsrMem(weight_md, &weight_tensor); in Compute()
250 src.SetUsrMem(src_md, &src_tensor); in Compute()
279 weight.SetUsrMem(weight_md, &weight_tensor); in Compute()
|
| D | mkl_maxpooling_op.cc | 150 dnn_data_output.SetUsrMem(pooling_fwd->GetPoolingFwdPd()->dst_desc(), in Compute()
218 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor); in AllocateWorkspaceTensor()
320 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor); in Compute()
|
| D | mkl_transpose_op.cc | 140 in.SetUsrMem(in_dims, in_strides, &in_tensor); in MKLTransposeND()
143 out.SetUsrMem(in_dims, out_strides, out_tensor); in MKLTransposeND()
|
| D | mkl_aggregate_ops.cc | 212 src.SetUsrMem(md, &src_tensor); in Compute()
220 dst.SetUsrMem(output_pd); in Compute()
|
| D | mkl_quantize_op.cc | 396 src.SetUsrMem(src_md, &min_shifted_input_tensor); in Compute()
398 src.SetUsrMem(src_md, &src_tensor); in Compute()
423 dst.SetUsrMem(dst_md, output_tensor); in Compute()
|
| D | mkl_slice_op.cc | 415 src.SetUsrMem(input_md, &input_tensor); in ComputeMklSlice()
429 src.SetUsrMem(input_md, &input_tensor); in ComputeMklSlice()
447 output.SetUsrMem(output_md, output_tensor); in ComputeMklSlice()
|
| D | mkl_concat_op.cc | 611 srcs[k].SetUsrMem(src_md, &input_tensors[k]); in Compute()
622 srcs[k].SetUsrMem(src_md, &input_tensors[k]); in Compute()
652 srcs[k].SetUsrMem(src_md, &input_tensors[k]); in Compute()
739 dst.SetUsrMem(dst_md, dst_tensor); in Compute()
773 dst.SetUsrMem(dst_md, dst_tensor); in Compute()
|
| D | mkl_tfconv_op.h | 97 input.SetUsrMem(input_mkl_md, &input_tensor); in ConvertMklToTf()
|
| D | mkl_matmul_op_fused.cc | 217 src_mkl.SetUsrMem(src_md, src_data); in Compute()
244 weight_mkl.SetUsrMem(weight_md, weight_data); in Compute()
|
| D | mkl_reshape_op.cc | 155 dnn_data_input.SetUsrMem(input_mkl_md, &input_tensor); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 542 src.SetUsrMem(fwd_src_md, &src_tensor); in Compute()
553 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
574 diff_filter.SetUsrMem(diff_filter_dims, diff_filter_format, in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 796 wksp.SetUsrMem(workspace_md, reserved_space_tensor); in Compute()
838 src.SetUsrMem(src_md, &src_tensor); in Compute()
1194 reorder_diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
1201 reorder_src.SetUsrMem(src_md, &src_tensor); in Compute()
1228 diff_dst.SetUsrMem(diff_dst_md, diff_dst_data); in Compute()
1235 src.SetUsrMem(src_md, src_data); in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 463 filter.SetUsrMem(fwd_filter_md, &filter_tensor); in Compute()
475 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
|
| D | mkl_relu_op.cc | 487 src.SetUsrMem(src_md, &src_tensor); in Compute()
692 src.SetUsrMem(src_md, &src_tensor); in Compute()
701 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
|
| D | mkl_avgpooling_op.cc | 267 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor); in Compute()
|
| D | mkl_conv_ops.cc | 615 src.SetUsrMem(src_md, &src_tensor); in Compute()
629 filter.SetUsrMem(filter_md, &filter_tensor); in Compute()
682 src.SetUsrMem(src_md, &src_tensor); in Compute()
707 filter.SetUsrMem(filter_md, &filter_tensor); in Compute()
1105 filter.SetUsrMem(filter_md, &filter_tensor); in CacheFilter()
|
| D | mkl_pooling_ops_common.h | 625 dnn_data_input->SetUsrMem(input_md, &input_tensor); in ConfigureInput()
|
| D | mkl_matmul_ops_common.h | 439 weight.SetUsrMem(weight_md, &weight_tensor); in CacheWeight()
|