Searched refs:fwd_cpu_stream (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_softmax_op.cc | 60 std::shared_ptr<stream> fwd_cpu_stream) { in Execute() argument
63 static_cast<void*>(const_cast<T*>(src_data)), *fwd_cpu_stream); in Execute()
65 *fwd_cpu_stream); in Execute()
73 execute_primitives(context_.fwd_primitives, fwd_cpu_stream, in Execute()
300 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
301 fwd_cpu_stream.reset(CreateStream(context, softmax_fwd->GetEngine())); in Compute()
302 softmax_fwd->Execute(src_data, dst_data, fwd_cpu_stream); in Compute()
|
| D | mkl_aggregate_ops.cc | 177 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
178 fwd_cpu_stream.reset(CreateStream(ctx, cpu_engine)); in Compute()
213 src.SetUsrMemDataHandle(&src_tensor, fwd_cpu_stream); in Compute()
235 dst.SetUsrMemDataHandle(dst_tensor, fwd_cpu_stream); in Compute()
245 sum_op.execute(*fwd_cpu_stream, net_args); in Compute()
|
| D | mkl_maxpooling_op.cc | 155 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
156 fwd_cpu_stream.reset(CreateStream(context, pooling_fwd->GetEngine())); in Compute()
160 pooling_fwd->Execute(src_data, dst_data, nullptr, fwd_cpu_stream); in Compute()
187 pooling_fwd->Execute(src_data, dst_data, ws_data, fwd_cpu_stream); in Compute()
|
| D | mkl_concat_op.cc | 734 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
735 fwd_cpu_stream.reset(CreateStream(context, cpu_engine)); in Compute()
740 dst.SetUsrMemDataHandle(dst_tensor, fwd_cpu_stream); in Compute()
748 concat_op.execute(*fwd_cpu_stream, net_args); in Compute()
771 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
772 fwd_cpu_stream.reset(CreateStream(context, concat_fwd->GetEngine())); in Compute()
774 dst.SetUsrMemDataHandle(dst_tensor, fwd_cpu_stream); in Compute()
777 fwd_cpu_stream); in Compute()
|
| D | mkl_avgpooling_op.cc | 131 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
132 fwd_cpu_stream.reset(CreateStream(context, pooling_fwd->GetEngine())); in Compute()
134 pooling_fwd->Execute(src_data, dst_data, nullptr, fwd_cpu_stream); in Compute()
|
| D | mkl_relu_op.cc | 481 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
482 fwd_cpu_stream.reset(CreateStream(context, eltwise_fwd->GetEngine())); in Compute()
537 eltwise_fwd->Execute(src_data, dst_data, fwd_cpu_stream); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 868 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
869 fwd_cpu_stream.reset(CreateStream(context, bn_fwd->GetEngine())); in Compute()
871 variance_op_data, fwd_cpu_stream, ws_data); in Compute()
|
| D | mkl_conv_ops.cc | 726 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
727 fwd_cpu_stream.reset(CreateStream(context, conv_fwd->GetEngine())); in Compute()
733 fwd_cpu_stream); in Compute()
735 conv_fwd->Execute(src_data, filter_data, dst_data, fwd_cpu_stream); in Compute()
|