| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_softmax_op.cc | 89 src.SetUsrMem(src_md, &src_tensor);
124 dst.SetUsrMem(src_md, output_tensor);
|
| mkl_conv_ops.h | 451 // Conv2DBackpropInput, we skip SetUsrMem call for input tensor (for
457 input.SetUsrMem(fwd_input_md, &input_tensor);
460 filter.SetUsrMem(fwd_filter_md, &filter_tensor);
468 outbackprop.SetUsrMem(outbprop_md, &outbprop_tensor);
472 outbackprop.SetUsrMem(outbprop_dims, tf_fmt, &outbprop_tensor);
478 output.SetUsrMem(bwd_output_dims, bwd_output_format);
|
| mkl_aggregate_ops.cc | 414 src1.SetUsrMem(md1, &src1_tensor);
415 src2.SetUsrMem(md2, &src2_tensor);
426 dst.SetUsrMem(common_md);
431 dst.SetUsrMem(common_md);
|
| mkl_tfconv_op.h | 99 input.SetUsrMem(input_mkl_md, &input_tensor);
|
| mkl_maxpooling_op.cc | 539 dnn_data_output.SetUsrMem(
544 dnn_data_output.SetUsrMem(output_dims_mkl_order,
603 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor);
660 output_dnn_data.SetUsrMem(original_input_md);
738 workspace_dnn_data->SetUsrMem(workspace_pd, &workspace_tensor);
[all...] |
| mkl_relu_op.cc | 411 src.SetUsrMem(src_md, &src_tensor);
443 dst.SetUsrMem(dst_md, dst_tensor);
552 src.SetUsrMem(src_md, &src_tensor);
553 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor);
600 diff_src.SetUsrMem(common_md, diff_src_tensor);
[all...] |
| mkl_input_conversion_op.cc | 334 input.SetUsrMem(input0_md, &input_tensor_0);
443 tf_input.SetUsrMem(input_tf_md, tf_tensor);
|
| mkl_pooling_ops_common.h | 208 dnn_data_input->SetUsrMem(input_md, &input_tensor);
377 input_gradient_dnn_data->SetUsrMem(original_input_grad_md,
|
| mkl_conv_ops.cc | 577 src.SetUsrMem(src_md, &src_tensor);
584 filter.SetUsrMem(filter_md, &filter_tensor);
590 output.SetUsrMem(output_dims_mkl_order, tf_fmt);
603 bias.SetUsrMem(bias_size, memory::format::x, &bias_tensor);
[all...] |
| mkl_fused_batch_norm_op.cc | [all...] |
| mkl_lrn_op.cc | 803 src_dnn_data.SetUsrMem(src_md, &src_tensor);
807 dst_dnn_data.SetUsrMem(src_md);
[all...] |
| mkl_reshape_op.cc | 246 dnn_data_input.SetUsrMem(input_mkl_md, &input_tensor);
|
| mkl_avgpooling_op.cc | 493 dnn_data_output.SetUsrMem(
499 dnn_data_output.SetUsrMem(output_dims_mkl_order,
595 output_diff_src.SetUsrMem(
|
| mkl_concat_op.cc | 704 srcs[k].SetUsrMem(src_md, &input_tensors[k]);
768 dst.SetUsrMem(dst_md, dst_tensor);
[all...] |
| mkl_conv_grad_filter_ops.cc | 554 bias_grad->SetUsrMem(bias_grad_md, bias_grad_tensor);
|
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util_test.cc | 66 a.SetUsrMem(dim1, strides1);
78 b.SetUsrMem(dim2, strides2);
|
| mkl_util.h | [all...] |
