Home
last modified time | relevance | path

Searched refs:bwd_cpu_stream (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_avgpooling_op.cc252 std::shared_ptr<stream> bwd_cpu_stream; in Compute() local
253 bwd_cpu_stream.reset(CreateStream(context, pooling_bwd->GetEngine())); in Compute()
281 bwd_cpu_stream); in Compute()
Dmkl_maxpooling_op.cc306 std::shared_ptr<stream> bwd_cpu_stream; in Compute() local
307 bwd_cpu_stream.reset(CreateStream(context, pooling_bwd->GetEngine())); in Compute()
337 bwd_cpu_stream); in Compute()
Dmkl_conv_grad_filter_ops.cc587 std::shared_ptr<stream> bwd_cpu_stream; in Compute() local
588 bwd_cpu_stream.reset(CreateStream(context, conv_bwd_filter->GetEngine())); in Compute()
593 diff_dst_data, bwd_cpu_stream); in Compute()
596 bwd_cpu_stream); in Compute()
Dmkl_conv_grad_input_ops.cc484 std::shared_ptr<stream> bwd_cpu_stream; in Compute() local
485 bwd_cpu_stream.reset(CreateStream(context, conv_bwd_input->GetEngine())); in Compute()
488 bwd_cpu_stream); in Compute()
Dmkl_relu_op.cc687 std::shared_ptr<stream> bwd_cpu_stream; in Compute() local
688 bwd_cpu_stream.reset(CreateStream(context, eltwise_bwd->GetEngine())); in Compute()
740 bwd_cpu_stream); in Compute()
Dmkl_fused_batch_norm_op.cc1274 std::shared_ptr<stream> bwd_cpu_stream; in Compute() local
1275 bwd_cpu_stream.reset(CreateStream(context, bn_bwd->GetEngine())); in Compute()
1278 res_space_data, bwd_cpu_stream); in Compute()