Searched refs:diff_src_data (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_grad_input_ops.cc | 106 void Execute(const T* diff_src_data, const T* filter_data, in Execute() argument
112 static_cast<T*>(const_cast<T*>(diff_src_data)), *bwd_input_stream); in Execute()
119 static_cast<T*>(const_cast<T*>(diff_src_data))); in Execute()
361 auto diff_src_data = diff_src_tensor->flat<T>().data(); in Compute() local
363 diff_src_data[i] = static_cast<T>(0); in Compute()
456 T* diff_src_data = in Compute() local
487 conv_bwd_input->Execute(diff_src_data, filter_data, diff_dst_data, in Compute()
|
| D | mkl_pooling_ops_common.cc | 189 T* diff_src_data, const void* ws_data, in Execute() argument
194 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data), in Execute()
203 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data)); in Execute()
|
| D | mkl_avgpooling_op.cc | 277 T* diff_src_data = output_tensor->flat<T>().data(); in Compute() local
280 pooling_bwd->Execute(diff_dst_data, diff_src_data, nullptr, in Compute()
|
| D | mkl_maxpooling_op.cc | 333 T* diff_src_data = output_tensor->flat<T>().data(); in Compute() local
336 pooling_bwd->Execute(diff_dst_data, diff_src_data, ws_data, in Compute()
|
| D | mkl_relu_op.cc | 256 void Execute(const T* src_data, const T* diff_dst_data, T* diff_src_data, in Execute() argument
263 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data), in Execute()
270 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data)); in Execute()
736 T* diff_src_data = diff_src_tensor->flat<T>().data(); in Compute() local
739 eltwise_bwd->Execute(src_data, diff_dst_data, diff_src_data, in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 422 const T* diff_dst_data, const U* weights_data, T* diff_src_data, in Execute() argument
443 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data), in Execute()
462 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data)); in Execute()
1265 T* diff_src_data = static_cast<T*>(diff_src_tensor->flat<T>().data()); in Compute() local
1277 weights_data, diff_src_data, diff_weights_data, in Compute()
1325 auto diff_src_data = (*diff_src_tensor)->flat<T>().data(); in HandleEmptyInput() local
1326 std::fill_n(diff_src_data, (*diff_src_tensor)->shape().num_elements(), in HandleEmptyInput()
|
| D | mkl_pooling_ops_common.h | 225 void Execute(const T* diff_dst_data, T* diff_src_data, const void* ws_data,
|