Searched refs:cpu_stream (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 146 std::shared_ptr<stream> cpu_stream; in Compute() local
147 cpu_stream.reset(CreateStream(ctx, matmul_prim->GetEngine())); in Compute()
149 out->flat<Scalar>().data(), cpu_stream); in Compute()
|
| D | mkl_matmul_op_fused.cc | 251 std::shared_ptr<stream> cpu_stream; in Compute() local
252 cpu_stream.reset(CreateStream(ctx, matmul_prim->GetEngine())); in Compute()
255 cpu_stream); in Compute()
|
| D | mkl_qmatmul_op.cc | 291 std::shared_ptr<stream> cpu_stream; in Compute() local
292 cpu_stream.reset(CreateStream(context, matmul_fwd->GetEngine())); in Compute()
295 context, matmul_fwd_pd, bias_tensor, weight_tensor, cpu_stream); in Compute()
297 cpu_stream); in Compute()
|
| D | mkl_quantize_op.cc | 476 std::shared_ptr<stream> cpu_stream; in Compute() local
477 cpu_stream.reset(CreateStream(ctx, reorder_prim->GetEngine())); in Compute()
479 cpu_stream); in Compute()
|
| D | mkl_matmul_ops_common.h | 720 std::shared_ptr<stream> cpu_stream; variable
721 cpu_stream.reset(CreateStream(ctx, matmul_prim->GetEngine()));
722 matmul_prim->Execute(a, b, c, cpu_stream);
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 615 stream* cpu_stream = CreateStream(context, cpu_engine); variable
617 net.at(i).execute(*cpu_stream, net_args->at(i));
619 cpu_stream->wait();
620 delete cpu_stream; variable
1497 std::shared_ptr<stream> cpu_stream; variable
1498 cpu_stream.reset(CreateStream(context, prim->GetEngine()));
1504 execute_primitives(net, cpu_stream, net_args);
1558 std::shared_ptr<stream> cpu_stream; variable
1559 cpu_stream.reset(CreateStream(context, prim->GetEngine()));
1565 execute_primitives(net, cpu_stream, net_args);
[all …]
|