| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_grad_input_ops.cc | 94 // diff_src_data: output data buffer of diff_src
98 void Execute(const T* diff_src_data, const T* filter_data,
101 static_cast<T*>(const_cast<T*>(diff_src_data)));
352 auto diff_src_data = diff_src_tensor->flat<T>().data(); local
354 diff_src_data[i] = 0;
435 T* diff_src_data = local
461 conv_bwd_input->Execute(diff_src_data, filter_data, diff_dst_data);
|
| mkl_fused_batch_norm_op.cc | 304 // diff_src_data: output data buffer of diff_src
307 const T* diff_dst_data, const T* weights_data, T* diff_src_data,
325 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data));
1009 T* diff_src_data = static_cast<T*>(diff_src_tensor->flat<T>().data()); variable
1061 auto diff_src_data = (*diff_src_tensor)->flat<T>().data(); local
[all...] |
| mkl_pooling_ops_common.cc | 200 T* diff_src_data, const void* ws_data) {
203 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data));
|
| mkl_relu_op.cc | 263 // diff_src_data: output data buffer of diff_src
264 void Execute(const T* src_data, const T* diff_dst_data, T* diff_src_data) {
269 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data));
696 T* diff_src_data = diff_src_tensor->flat<T>().data(); local
699 eltwise_bwd->Execute(src_data, diff_dst_data, diff_src_data);
[all...] |
| mkl_avgpooling_op.cc | 652 T* diff_src_data = output_tensor->flat<T>().data(); variable
655 pooling_bwd->Execute(diff_dst_data, diff_src_data);
|
| mkl_maxpooling_op.cc | 781 T* diff_src_data = output_tensor->flat<T>().data(); variable
[all...] |
| mkl_pooling_ops_common.h | 223 // diff_src_data: output data buffer of diff_src
225 void Execute(const T* diff_dst_data, T* diff_src_data,
|
