| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_grad_input_ops.cc | 96 // diff_dst_data: input data buffer of dst
99 const T* diff_dst_data) {
105 static_cast<T*>(const_cast<T*>(diff_dst_data)));
450 T* diff_dst_data = nullptr; local
454 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle());
456 diff_dst_data =
461 conv_bwd_input->Execute(diff_src_data, filter_data, diff_dst_data);
|
| mkl_conv_grad_filter_ops.cc | 97 // diff_dst_data: input data buffer of diff_dst
99 const T* diff_bias_data, const T* diff_dst_data) {
107 static_cast<void*>(const_cast<T*>(diff_dst_data)));
121 // diff_dst_data: input data buffer of diff_dst
123 const T* diff_dst_data) {
129 static_cast<void*>(const_cast<T*>(diff_dst_data)));
547 T* diff_dst_data = nullptr; local
552 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle());
554 diff_dst_data =
582 diff_dst_data);
[all...] |
| mkl_fused_batch_norm_op.cc | 302 // diff_dst_data: input data buffer of diff_dst
307 const T* diff_dst_data, const T* weights_data, T* diff_src_data,
316 static_cast<void*>(const_cast<T*>(diff_dst_data)));
975 const T* diff_dst_data = diff_dst_tensor.flat<T>().data(); variable
[all...] |
| mkl_relu_op.cc | 262 // diff_dst_data: input data buffer of diff_dst
264 void Execute(const T* src_data, const T* diff_dst_data, T* diff_src_data) {
268 static_cast<void*>(const_cast<T*>(diff_dst_data)));
657 const T* diff_dst_data = diff_dst_tensor.flat<T>().data(); local
662 diff_dst_data = const_cast<T*>(
699 eltwise_bwd->Execute(src_data, diff_dst_data, diff_src_data);
[all...] |
| mkl_pooling_ops_common.cc | 199 void MklPoolingBwdPrimitive<T>::Execute(const T* diff_dst_data,
202 static_cast<void*>(const_cast<T*>(diff_dst_data)));
|
| mkl_avgpooling_op.cc | 641 const T* diff_dst_data = grad_tensor.flat<T>().data(); variable
648 diff_dst_data = const_cast<T*>(
655 pooling_bwd->Execute(diff_dst_data, diff_src_data);
|
| mkl_maxpooling_op.cc | 752 const T* diff_dst_data = grad_tensor.flat<T>().data(); variable
759 diff_dst_data = const_cast<T*>(
[all...] |
| mkl_pooling_ops_common.h | 222 // diff_dst_data: input data buffer of diff_dst
225 void Execute(const T* diff_dst_data, T* diff_src_data,
|
