Searched refs:fwd_stream (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.cc | 88 std::shared_ptr<stream> fwd_stream) { in Execute() argument
91 static_cast<void*>(const_cast<T*>(src_data)), *fwd_stream); in Execute()
92 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data), *fwd_stream); in Execute()
97 context_.ws_mem->set_data_handle(ws_data, *fwd_stream); in Execute()
110 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
|
| D | mkl_matmul_ops_common.h | 103 std::shared_ptr<stream> fwd_stream) { in Execute() argument
106 static_cast<void*>(const_cast<Tinput*>(src_data)), *fwd_stream); in Execute()
108 static_cast<void*>(const_cast<Tweight*>(weight_data)), *fwd_stream); in Execute()
112 *fwd_stream); in Execute()
123 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
|
| D | mkl_fused_batch_norm_op.cc | 80 std::shared_ptr<stream> fwd_stream, U* workspace_data) { in Execute() argument
84 static_cast<void*>(const_cast<T*>(src_data)), *fwd_stream); in Execute()
86 *fwd_stream); in Execute()
90 static_cast<void*>(const_cast<U*>(weights_data)), *fwd_stream); in Execute()
95 *fwd_stream); in Execute()
97 *fwd_stream); in Execute()
100 context_.ws_mem->set_data_handle(workspace_data, *fwd_stream); in Execute()
122 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
|
| D | mkl_concat_op.cc | 280 std::shared_ptr<stream> fwd_stream) { in Execute() argument
285 static_cast<void*>(in_data[i].get_data_handle()), *fwd_stream); in Execute()
288 static_cast<void*>(dst_data.get_data_handle()), *fwd_stream); in Execute()
301 execute_primitives(context_.fwd_primitives, fwd_stream,
|
| D | mkl_conv_ops.cc | 116 std::shared_ptr<stream> fwd_stream) { in Execute() argument
120 static_cast<void*>(const_cast<Tinput*>(src_data)), *fwd_stream); in Execute()
122 static_cast<void*>(const_cast<Tfilter*>(filter_data)), *fwd_stream); in Execute()
125 static_cast<void*>(const_cast<Tbias*>(bias_data)), *fwd_stream); in Execute()
128 static_cast<void*>(const_cast<Toutput*>(dst_data)), *fwd_stream); in Execute()
145 context_.fwd_primitives.at(i).execute(*fwd_stream, in Execute()
163 const Toutput* dst_data, std::shared_ptr<stream> fwd_stream) { in Execute() argument
164 Execute(src_data, filter_data, nullptr, dst_data, fwd_stream); in Execute()
|
| D | mkl_relu_op.cc | 76 std::shared_ptr<stream> fwd_stream) { in Execute() argument
79 static_cast<void*>(const_cast<T*>(src_data)), *fwd_stream); in Execute()
81 *fwd_stream); in Execute()
89 execute_primitives(context_.fwd_primitives, fwd_stream, in Execute()
|
| D | mkl_pooling_ops_common.h | 87 std::shared_ptr<stream> fwd_stream);
131 std::shared_ptr<mkldnn::stream> fwd_stream; member
|