Home
last modified time | relevance | path

Searched refs:diff_dst_data (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_filter_ops.cc105 const T* diff_bias_data, const T* diff_dst_data, in Execute() argument
120 static_cast<void*>(const_cast<T*>(diff_dst_data)), *bwd_filter_stream); in Execute()
131 static_cast<void*>(const_cast<T*>(diff_dst_data))); in Execute()
149 const T* diff_dst_data, in Execute() argument
151 Execute(src_data, diff_filter_data, nullptr, diff_dst_data, in Execute()
550 T* diff_dst_data = nullptr; in Compute() local
556 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle()); in Compute()
558 diff_dst_data = in Compute()
593 diff_dst_data, bwd_cpu_stream); in Compute()
595 conv_bwd_filter->Execute(src_data, diff_filter_data, diff_dst_data, in Compute()
Dmkl_conv_grad_input_ops.cc107 const T* diff_dst_data, in Execute() argument
116 static_cast<T*>(const_cast<T*>(diff_dst_data)), *bwd_input_stream); in Execute()
123 static_cast<T*>(const_cast<T*>(diff_dst_data))); in Execute()
472 T* diff_dst_data = nullptr; in Compute() local
478 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle()); in Compute()
480 diff_dst_data = in Compute()
487 conv_bwd_input->Execute(diff_src_data, filter_data, diff_dst_data, in Compute()
Dmkl_avgpooling_op.cc264 T* diff_dst_data = nullptr; in Compute() local
270 diff_dst_data = in Compute()
273 diff_dst_data = in Compute()
280 pooling_bwd->Execute(diff_dst_data, diff_src_data, nullptr, in Compute()
Dmkl_maxpooling_op.cc317 T* diff_dst_data = nullptr; in Compute() local
323 diff_dst_data = in Compute()
326 diff_dst_data = in Compute()
336 pooling_bwd->Execute(diff_dst_data, diff_src_data, ws_data, in Compute()
Dmkl_pooling_ops_common.cc188 void MklPoolingBwdPrimitive<T>::Execute(const T* diff_dst_data, in Execute() argument
193 static_cast<void*>(const_cast<T*>(diff_dst_data)), *bwd_stream); in Execute()
202 static_cast<void*>(const_cast<T*>(diff_dst_data))); in Execute()
Dmkl_relu_op.cc256 void Execute(const T* src_data, const T* diff_dst_data, T* diff_src_data, in Execute() argument
262 static_cast<void*>(const_cast<T*>(diff_dst_data)), *bwd_stream); in Execute()
269 static_cast<void*>(const_cast<T*>(diff_dst_data))); in Execute()
699 const T* diff_dst_data = diff_dst_tensor.flat<T>().data(); in Compute() local
704 diff_dst_data = const_cast<T*>( in Compute()
739 eltwise_bwd->Execute(src_data, diff_dst_data, diff_src_data, in Compute()
Dmkl_fused_batch_norm_op.cc422 const T* diff_dst_data, const U* weights_data, T* diff_src_data, in Execute() argument
434 static_cast<void*>(const_cast<T*>(diff_dst_data)), *bwd_stream); in Execute()
453 static_cast<void*>(const_cast<T*>(diff_dst_data))); in Execute()
1183 T* diff_dst_data = in Compute() local
1197 diff_dst_data = in Compute()
1228 diff_dst.SetUsrMem(diff_dst_md, diff_dst_data); in Compute()
1231 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle()); in Compute()
1276 bn_bwd->Execute(src_data, mean_data, variance_data, diff_dst_data, in Compute()
Dmkl_pooling_ops_common.h225 void Execute(const T* diff_dst_data, T* diff_src_data, const void* ws_data,