Home
last modified time | relevance | path

Searched refs:CheckReorderToOpMem (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_input_conversion_op.cc146 CHECK_EQ(input.CheckReorderToOpMem( in Compute()
245 bool reordered = tf_input.CheckReorderToOpMem( in Compute()
Dmkl_aggregate_ops.cc207 src1.CheckReorderToOpMem(srcs_pd[0]); in Compute()
208 src2.CheckReorderToOpMem(srcs_pd[1]); in Compute()
Dmkl_tfconv_op.h107 CHECK_EQ(input.CheckReorderToOpMem(output_tf_pd, output_tensor), true); in ConvertMklToTf()
Dmkl_lrn_op.cc179 src_dnn_data->CheckReorderToOpMem(lrn_fwd_desc.src_primitive_desc()); in PrepareAndExecuteNet()
491 input_gradient_diff_dst->CheckReorderToOpMem( in PrepareAndExecuteNet()
495 src_dnn_data->CheckReorderToOpMem(lrn_fwd_desc.src_primitive_desc()); in PrepareAndExecuteNet()
Dmkl_reshape_op.cc166 if (dnn_data_input.CheckReorderToOpMem(output_tf_pd, output_tensor)) { in Compute()
Dmkl_maxpooling_op.cc602 dnn_data_input.CheckReorderToOpMem(src_target_primitive_desc); in Compute()
758 grad_dnn_data.CheckReorderToOpMem(target_diff_dst); in Compute()
777 workspace_dnn_data.CheckReorderToOpMem(target_ws); in Compute()
Dmkl_conv_grad_input_ops.cc443 filter.CheckReorderToOpMem(bwd_input_pd->weights_primitive_desc()); in Compute()
453 diff_dst.CheckReorderToOpMem(bwd_input_pd->diff_dst_primitive_desc()); in Compute()
Dmkl_conv_ops.cc1103 src.CheckReorderToOpMem(conv_fwd_pd.get()->src_primitive_desc()); in Compute()
1130 filter.CheckReorderToOpMem( in Compute()
1133 filter.CheckReorderToOpMem( in Compute()
1367 src->CheckReorderToOpMem(conv_prim_desc.src_primitive_desc()); in PrepareAndExecuteNet()
1371 filter->CheckReorderToOpMem(conv_prim_desc.weights_primitive_desc(), in PrepareAndExecuteNet()
1419 filter.CheckReorderToOpMem(conv_fwd_pd.get()->weights_primitive_desc()); in CacheFilter()
Dmkl_avgpooling_op.cc525 dnn_data_input.CheckReorderToOpMem(src_target_primitive_desc); in Compute()
647 grad_dnn_data.CheckReorderToOpMem(target_diff_dst); in Compute()
Dmkl_conv_grad_filter_ops.cc541 src.CheckReorderToOpMem(bwd_filter_pd->src_primitive_desc()); in Compute()
551 diff_dst.CheckReorderToOpMem(bwd_filter_pd->diff_dst_primitive_desc()); in Compute()
Dmkl_fused_batch_norm_op.cc653 src.CheckReorderToOpMem(src_target); in Compute()
970 src.CheckReorderToOpMem(src_target); in Compute()
983 diff_dst.CheckReorderToOpMem(diff_dst_target); in Compute()
Dmkl_relu_op.cc497 src.CheckReorderToOpMem(src_target_pd); in Compute()
651 src.CheckReorderToOpMem( in Compute()
660 diff_dst.CheckReorderToOpMem( in Compute()
Dmkl_concat_op.cc371 srcs[k].CheckReorderToOpMem(srcs_pd[k]); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h857 CHECK_EQ(input.CheckReorderToOpMem(output_tf_pd, &output_tensor, &net), in ConvertMklToTF()
1897 inline bool CheckReorderToOpMem(const memory::primitive_desc& op_pd, in CheckReorderToOpMem() function
1913 inline bool CheckReorderToOpMem(const memory::primitive_desc& op_pd) { in CheckReorderToOpMem() function
1938 inline bool CheckReorderToOpMem(const memory::primitive_desc& op_pd, in CheckReorderToOpMem() function
1956 inline bool CheckReorderToOpMem(const memory::primitive_desc& op_pd, in CheckReorderToOpMem() function
1983 inline bool CheckReorderToOpMem(const memory::primitive_desc& op_pd, in CheckReorderToOpMem() function
1988 return CheckReorderToOpMem(op_pd, GetTensorBuffer(reorder_tensor), net); in CheckReorderToOpMem()
1994 inline bool CheckReorderToOpMem(const memory::primitive_desc& op_pd, in CheckReorderToOpMem() function
1997 return CheckReorderToOpMem(op_pd, GetTensorBuffer(reorder_tensor)); in CheckReorderToOpMem()