| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_aggregate_ops.cc | 172 src1.SetUsrMem(md1, &src1_tensor);
173 src2.SetUsrMem(md2, &src2_tensor);
184 dst.SetUsrMem(common_md);
189 dst.SetUsrMem(common_md);
|
| mkl_softmax_op.cc | 127 src.SetUsrMem(src_md, &src_tensor);
160 dst.SetUsrMem(src_md, output_tensor);
|
| mkl_tfconv_op.h | 95 input.SetUsrMem(input_mkl_md, &input_tensor);
|
| mkl_input_conversion_op.cc | 142 input.SetUsrMem(input0_md, &input_tensor_0);
241 tf_input.SetUsrMem(input_tf_md, tf_tensor);
|
| mkl_transpose_op.cc | 138 in.SetUsrMem(in_dims, in_strides, &in_tensor);
141 out.SetUsrMem(in_dims, out_strides, out_tensor);
|
| mkl_lrn_op.cc | 133 src_dnn_data.SetUsrMem(src_md, &src_tensor);
137 dst_dnn_data.SetUsrMem(src_md);
285 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor);
388 orig_input_dnn_data.SetUsrMem(orig_input_md, &orig_input_tensor);
392 output_dnn_data.SetUsrMem(orig_input_md);
478 input_grad_dnn_data->SetUsrMem(input_grad_md, &input_grad_tensor);
516 workspace_dnn_data->SetUsrMem(workspace_pd, &workspace_tensor);
|
| mkl_concat_op.cc | 306 srcs[k].SetUsrMem(src_md, &input_tensors[k]);
317 srcs[k].SetUsrMem(src_md, &input_tensors[k]);
341 srcs[k].SetUsrMem(src_md, &input_tensors[k]);
415 dst.SetUsrMem(dst_md, dst_tensor);
|
| mkl_maxpooling_op.cc | 591 dnn_data_output.SetUsrMem(output_dims_mkl_order,
598 dnn_data_input.SetUsrMem(input_md, &input_tensor);
671 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor);
757 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor);
[all...] |
| mkl_slice_op.cc | 401 src.SetUsrMem(input_md, &input_tensor);
411 src.SetUsrMem(input_md, &input_tensor);
423 output.SetUsrMem(output_md, output_tensor);
|
| mkl_reshape_op.cc | 148 dnn_data_input.SetUsrMem(input_mkl_md, &input_tensor);
|
| mkl_conv_grad_input_ops.cc | 442 filter.SetUsrMem(fwd_filter_md, &filter_tensor);
452 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor);
|
| mkl_conv_grad_filter_ops.cc | 540 src.SetUsrMem(fwd_src_md, &src_tensor);
550 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor);
565 diff_filter.SetUsrMem(bwd_output_dims, GetOutputFormat(tf_fmt),
|
| mkl_fused_batch_norm_op.cc | 647 src.SetUsrMem(src_md, &src_tensor);
[all...] |
| mkl_relu_op.cc | 493 src.SetUsrMem(src_md, &src_tensor);
650 src.SetUsrMem(src_md, &src_tensor);
659 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor);
[all...] |
| mkl_avgpooling_op.cc | 521 dnn_data_input.SetUsrMem(input_md, &input_tensor);
646 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor);
|
| mkl_pooling_ops_common.h | 623 dnn_data_input->SetUsrMem(input_md, &input_tensor);
732 input_gradient_dnn_data->SetUsrMem(original_input_grad_md,
|
| mkl_conv_ops.cc | [all...] |
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util_test.cc | 66 a.SetUsrMem(dim1, strides1);
78 b.SetUsrMem(dim2, strides2);
|
| mkl_util.h | 852 input.SetUsrMem(input_mkl_md, &mkl_tensor);
[all...] |
