Home
last modified time | relevance | path

Searched refs:diff_dst (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc473 MklDnnData<T> diff_dst(&cpu_engine_); in Compute() local
475 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
476 diff_dst.CheckReorderToOpMem(bwd_input_pd.get()->diff_dst_desc(), in Compute()
478 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle()); in Compute()
Dmkl_conv_grad_filter_ops.cc551 MklDnnData<T> diff_dst(&cpu_engine_); in Compute() local
553 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
554 diff_dst.CheckReorderToOpMem(bwd_filter_pd->diff_dst_desc(), in Compute()
556 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle()); in Compute()
Dmkl_relu_op.cc588 MklDnnData<T> diff_dst(&cpu_engine); in Compute() local
701 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
702 diff_dst.CheckReorderToOpMem(eltwise_bwd_pd.get()->diff_src_desc(), in Compute()
705 reinterpret_cast<T*>(diff_dst.GetOpMem().get_data_handle())); in Compute()
Dmkl_fused_batch_norm_op.cc1157 MklDnnData<T> diff_dst(&cpu_engine_); in Compute() local
1228 diff_dst.SetUsrMem(diff_dst_md, diff_dst_data); in Compute()
1229 diff_dst.CheckReorderToOpMem(bn_bwd_pd->diff_dst_desc(), cpu_engine_, in Compute()
1231 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle()); in Compute()