Searched refs:CheckReorderToOpMem (Results 1 – 14 of 14) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | mkl_input_conversion_op.cc | 146 CHECK_EQ(input.CheckReorderToOpMem( in Compute() 245 bool reordered = tf_input.CheckReorderToOpMem( in Compute()
|
D | mkl_aggregate_ops.cc | 207 src1.CheckReorderToOpMem(srcs_pd[0]); in Compute() 208 src2.CheckReorderToOpMem(srcs_pd[1]); in Compute()
|
D | mkl_tfconv_op.h | 107 CHECK_EQ(input.CheckReorderToOpMem(output_tf_pd, output_tensor), true); in ConvertMklToTf()
|
D | mkl_lrn_op.cc | 179 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()
|
D | mkl_reshape_op.cc | 166 if (dnn_data_input.CheckReorderToOpMem(output_tf_pd, output_tensor)) { in Compute()
|
D | mkl_maxpooling_op.cc | 602 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()
|
D | mkl_conv_grad_input_ops.cc | 443 filter.CheckReorderToOpMem(bwd_input_pd->weights_primitive_desc()); in Compute() 453 diff_dst.CheckReorderToOpMem(bwd_input_pd->diff_dst_primitive_desc()); in Compute()
|
D | mkl_conv_ops.cc | 1103 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()
|
D | mkl_avgpooling_op.cc | 525 dnn_data_input.CheckReorderToOpMem(src_target_primitive_desc); in Compute() 647 grad_dnn_data.CheckReorderToOpMem(target_diff_dst); in Compute()
|
D | mkl_conv_grad_filter_ops.cc | 541 src.CheckReorderToOpMem(bwd_filter_pd->src_primitive_desc()); in Compute() 551 diff_dst.CheckReorderToOpMem(bwd_filter_pd->diff_dst_primitive_desc()); in Compute()
|
D | mkl_fused_batch_norm_op.cc | 653 src.CheckReorderToOpMem(src_target); in Compute() 970 src.CheckReorderToOpMem(src_target); in Compute() 983 diff_dst.CheckReorderToOpMem(diff_dst_target); in Compute()
|
D | mkl_relu_op.cc | 497 src.CheckReorderToOpMem(src_target_pd); in Compute() 651 src.CheckReorderToOpMem( in Compute() 660 diff_dst.CheckReorderToOpMem( in Compute()
|
D | mkl_concat_op.cc | 371 srcs[k].CheckReorderToOpMem(srcs_pd[k]); in Compute()
|
/external/tensorflow/tensorflow/core/util/ |
D | mkl_util.h | 857 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()
|