Home
last modified time | relevance | path

Searched refs:GetOpMem (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_lrn_op.cc173 net_args.push_back({{MKLDNN_ARG_SRC, src_dnn_data.GetOpMem()}, in Compute()
174 {MKLDNN_ARG_WORKSPACE, workspace_dnn_data.GetOpMem()}, in Compute()
175 {MKLDNN_ARG_DST, dst_dnn_data.GetOpMem()}}); in Compute()
437 net_args.push_back({{MKLDNN_ARG_SRC, orig_input_dnn_data.GetOpMem()}, in Compute()
438 {MKLDNN_ARG_DIFF_DST, input_grad_dnn_data.GetOpMem()}, in Compute()
439 {MKLDNN_ARG_DST, output_dnn_data.GetOpMem()}}); in Compute()
Dmkl_concat_op.cc614 inputs.push_back(srcs[k].GetOpMem()); in Compute()
655 inputs.push_back(srcs[k].GetOpMem()); in Compute()
658 srcs_mem.push_back(srcs[k].GetOpMem()); in Compute()
700 inputs.push_back(srcs[k].GetOpMem()); in Compute()
744 {MKLDNN_ARG_DST, dst.GetOpMem()}}; in Compute()
776 concat_fwd->Execute(srcs_mem, dst.GetOpMem(), concat_fwd_dims, in Compute()
Dmkl_aggregate_ops.cc214 inputs.push_back(src.GetOpMem()); in Compute()
241 {MKLDNN_ARG_DST, dst.GetOpMem()}}; in Compute()
Dmkl_matmul_op_fused.cc220 src_data = reinterpret_cast<T*>(src_mkl.GetOpMem().get_data_handle()); in Compute()
248 reinterpret_cast<T*>(weight_mkl.GetOpMem().get_data_handle()); in Compute()
Dmkl_conv_ops.cc684 src_data = static_cast<Tinput*>(src.GetOpMem().get_data_handle()); in Compute()
718 static_cast<Tfilter*>(filter.GetOpMem().get_data_handle()); in Compute()
1063 net_args.push_back({{MKLDNN_ARG_SRC, src->GetOpMem()}, in PrepareAndExecuteNet()
1064 {MKLDNN_ARG_WEIGHTS, filter->GetOpMem()}, in PrepareAndExecuteNet()
1065 {MKLDNN_ARG_BIAS, bias->GetOpMem()}, in PrepareAndExecuteNet()
1066 {MKLDNN_ARG_DST, output->GetOpMem()}}); in PrepareAndExecuteNet()
1070 net_args.push_back({{MKLDNN_ARG_SRC, src->GetOpMem()}, in PrepareAndExecuteNet()
1071 {MKLDNN_ARG_WEIGHTS, filter->GetOpMem()}, in PrepareAndExecuteNet()
1072 {MKLDNN_ARG_DST, output->GetOpMem()}}); in PrepareAndExecuteNet()
1108 filter_data = static_cast<Tfilter*>(filter.GetOpMem().get_data_handle()); in CacheFilter()
[all …]
Dmkl_maxpooling_op.cc185 static_cast<T*>(dnn_data_wksp.GetOpMem().get_data_handle()); in Compute()
324 static_cast<T*>(grad_dnn_data.GetOpMem().get_data_handle()); in Compute()
Dmkl_conv_grad_filter_ops.cc545 src_data = static_cast<T*>(src.GetOpMem().get_data_handle()); in Compute()
556 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle()); in Compute()
580 static_cast<T*>(diff_filter.GetOpMem().get_data_handle()); in Compute()
Dmkl_fused_batch_norm_op.cc797 ws_data = static_cast<U*>(wksp.GetOpMem().get_data_handle()); in Compute()
840 src_data = static_cast<T*>(src.GetOpMem().get_data_handle()); in Compute()
1198 static_cast<T*>(reorder_diff_dst.GetOpMem().get_data_handle()); in Compute()
1204 src_data = static_cast<T*>(reorder_src.GetOpMem().get_data_handle()); in Compute()
1231 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle()); in Compute()
1237 src_data = static_cast<T*>(src.GetOpMem().get_data_handle()); in Compute()
Dmkl_conv_grad_input_ops.cc466 filter_data = static_cast<T*>(filter.GetOpMem().get_data_handle()); in Compute()
478 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle()); in Compute()
Dmkl_qmatmul_op.cc253 src_data = static_cast<Tinput*>(src.GetOpMem().get_data_handle()); in Compute()
283 static_cast<Tweight*>(weight.GetOpMem().get_data_handle()); in Compute()
Dmkl_relu_op.cc491 reinterpret_cast<T*>(src.GetOpMem().get_data_handle())); in Compute()
696 reinterpret_cast<T*>(src.GetOpMem().get_data_handle())); in Compute()
705 reinterpret_cast<T*>(diff_dst.GetOpMem().get_data_handle())); in Compute()
Dmkl_avgpooling_op.cc271 static_cast<T*>(grad_dnn_data.GetOpMem().get_data_handle()); in Compute()
Dmkl_slice_op.cc450 MklSliceParams sliceParams(&src.GetOpMem(), output.GetUsrMem(), in ComputeMklSlice()
Dmkl_matmul_ops_common.h442 weight_data = static_cast<Tweight*>(weight.GetOpMem().get_data_handle()); in CacheWeight()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1419 inline const memory& GetOpMem() const { in GetOpMem() function