Searched refs:bwd_stream (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.cc | 190 std::shared_ptr<stream> bwd_stream) { in Execute() argument
193 static_cast<void*>(const_cast<T*>(diff_dst_data)), *bwd_stream); in Execute()
195 *bwd_stream); in Execute()
198 context_.ws_mem->set_data_handle(const_cast<void*>(ws_data), *bwd_stream); in Execute()
210 execute_primitives(context_.bwd_primitives, bwd_stream, context_.net_args); in Execute()
|
| D | mkl_fused_batch_norm_op.cc | 424 std::shared_ptr<stream> bwd_stream) { in Execute() argument
428 static_cast<void*>(const_cast<T*>(src_data)), *bwd_stream); in Execute()
430 static_cast<void*>(const_cast<U*>(mean_data)), *bwd_stream); in Execute()
432 static_cast<void*>(const_cast<U*>(variance_data)), *bwd_stream); in Execute()
434 static_cast<void*>(const_cast<T*>(diff_dst_data)), *bwd_stream); in Execute()
438 static_cast<void*>(const_cast<U*>(weights_data)), *bwd_stream); in Execute()
440 static_cast<void*>(diff_weights_data), *bwd_stream); in Execute()
444 *bwd_stream); in Execute()
466 execute_primitives(context_.bwd_primitives, bwd_stream, context_.net_args); in Execute()
|
| D | mkl_relu_op.cc | 257 std::shared_ptr<stream> bwd_stream) { in Execute() argument
260 static_cast<void*>(const_cast<T*>(src_data)), *bwd_stream); in Execute()
262 static_cast<void*>(const_cast<T*>(diff_dst_data)), *bwd_stream); in Execute()
264 *bwd_stream); in Execute()
274 execute_primitives(context_.bwd_primitives, bwd_stream, in Execute()
|
| D | mkl_pooling_ops_common.h | 226 std::shared_ptr<stream> bwd_stream);
274 std::shared_ptr<mkldnn::stream> bwd_stream; member
|