| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 149 input.CheckReorderToOpMem(input1_md, tensor_out, net, net_args, in Compute()
247 bool reordered = tf_input.CheckReorderToOpMem(output_mkl_md, tensor_out, in Compute()
|
| D | mkl_tfconv_op.h | 111 input.CheckReorderToOpMem(output_tf_md, output_tensor, context), in ConvertMklToTf()
|
| D | mkl_matmul_op_fused.cc | 218 src_mkl.CheckReorderToOpMem(matmul_pd.get()->src_desc(), in Compute()
245 weight_mkl.CheckReorderToOpMem(matmul_pd.get()->weights_desc(), in Compute()
|
| D | mkl_reshape_op.cc | 171 if (dnn_data_input.CheckReorderToOpMem(output_tf_md, output_tensor, in Compute()
|
| D | mkl_lrn_op.cc | 167 src_dnn_data.CheckReorderToOpMem(lrn_prim_desc.src_desc(), cpu_engine_); in Compute()
427 input_grad_dnn_data.CheckReorderToOpMem(lrn_bwd_prim_desc.diff_dst_desc(), in Compute()
431 orig_input_dnn_data.CheckReorderToOpMem(lrn_fwd_prim_desc.src_desc(), in Compute()
|
| D | mkl_qmatmul_op.cc | 251 src.CheckReorderToOpMem(matmul_fwd_pd.get()->src_desc(), in Compute()
280 weight.CheckReorderToOpMem(matmul_fwd_pd.get()->weights_desc(), in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 464 filter.CheckReorderToOpMem(bwd_input_pd.get()->weights_desc(), in Compute()
476 diff_dst.CheckReorderToOpMem(bwd_input_pd.get()->diff_dst_desc(), in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 839 src.CheckReorderToOpMem(bn_fwd_pd->src_desc(), cpu_engine_, context); in Compute()
1195 reorder_diff_dst.CheckReorderToOpMem(src_md, cpu_engine_, context); in Compute()
1202 reorder_src.CheckReorderToOpMem(diff_dst_md, cpu_engine_, context); in Compute()
1229 diff_dst.CheckReorderToOpMem(bn_bwd_pd->diff_dst_desc(), cpu_engine_, in Compute()
1236 src.CheckReorderToOpMem(bn_bwd_pd->src_desc(), cpu_engine_, context); in Compute()
|
| D | mkl_conv_ops.cc | 683 src.CheckReorderToOpMem(conv_fwd_pd->src_desc(), cpu_engine_, context); in Compute()
709 filter.CheckReorderToOpMem(conv_fwd_pd->weights_desc(), cpu_engine_, in Compute()
712 filter.CheckReorderToOpMem( in Compute()
1050 src->CheckReorderToOpMem(conv_prim_desc.src_desc(), cpu_engine_); in PrepareAndExecuteNet()
1054 filter->CheckReorderToOpMem(conv_prim_desc.weights_desc(), in PrepareAndExecuteNet()
1106 filter.CheckReorderToOpMem(conv_fwd_pd.get()->weights_desc(), in CacheFilter()
|
| D | mkl_conv_grad_filter_ops.cc | 543 src.CheckReorderToOpMem(bwd_filter_pd->src_desc(), cpu_engine_, in Compute()
554 diff_dst.CheckReorderToOpMem(bwd_filter_pd->diff_dst_desc(), in Compute()
|
| D | mkl_relu_op.cc | 488 src.CheckReorderToOpMem(eltwise_fwd_pd->src_desc(), cpu_engine, in Compute()
693 src.CheckReorderToOpMem(eltwise_bwd_pd.get()->diff_src_desc(), in Compute()
702 diff_dst.CheckReorderToOpMem(eltwise_bwd_pd.get()->diff_src_desc(), in Compute()
|
| D | mkl_avgpooling_op.cc | 268 grad_dnn_data.CheckReorderToOpMem(pooling_bwd_pd->diff_dst_desc(), in Compute()
|
| D | mkl_maxpooling_op.cc | 321 grad_dnn_data.CheckReorderToOpMem(pooling_bwd_pd->diff_dst_desc(), in Compute()
|
| D | mkl_slice_op.cc | 436 src.CheckReorderToOpMem(op_md, cpu_engine, context); in ComputeMklSlice()
|
| D | mkl_matmul_ops_common.h | 440 weight.CheckReorderToOpMem(matmul_fwd_pd.get()->weights_desc(), cpu_engine_, in CacheWeight()
|
| D | mkl_concat_op.cc | 699 srcs[k].CheckReorderToOpMem(srcs_pd[k], cpu_engine, context); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 650 bool status = input.CheckReorderToOpMem(output_tf_md, output_tf_tensor, in ConvertMklToTF()
1470 inline bool CheckReorderToOpMem(const memory::desc& op_md, in CheckReorderToOpMem() function
1487 inline bool CheckReorderToOpMem(const memory::desc& op_md,
1524 inline bool CheckReorderToOpMem(const memory::desc& op_md, in CheckReorderToOpMem() function
1546 inline bool CheckReorderToOpMem(const memory::desc& op_md,
1585 inline bool CheckReorderToOpMem(const memory::desc& op_md, in CheckReorderToOpMem() function
1591 return CheckReorderToOpMem(op_md, GetTensorBuffer(reorder_tensor), net, in CheckReorderToOpMem()
1599 inline bool CheckReorderToOpMem(const memory::desc& op_pd,
1603 return CheckReorderToOpMem(op_pd, GetTensorBuffer(reorder_tensor),
|