Home
last modified time | relevance | path

Searched defs:cpu_stream (Results 1 – 8 of 8) sorted by relevance

/external/pytorch/torch/csrc/jit/codegen/onednn/
DLlgaTensorImpl.cpp42 static dnnl::stream cpu_stream{Engine::getEngine()}; in getStream() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_batch_matmul_op.cc161 std::shared_ptr<stream> cpu_stream; in Compute() local
Dmkl_matmul_op_fused.cc265 std::shared_ptr<stream> cpu_stream; in Compute() local
Dmkl_einsum_op.cc121 std::shared_ptr<stream> cpu_stream; in MKLContractOperands() local
Dmkl_quantize_op.cc485 std::shared_ptr<stream> cpu_stream; in Compute() local
Dmkl_qmatmul_op.cc293 std::shared_ptr<stream> cpu_stream; in Compute() local
Dmkl_matmul_ops_common.h915 std::shared_ptr<stream> cpu_stream; variable
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h612 std::unique_ptr<stream> cpu_stream; variable
1542 std::shared_ptr<stream> cpu_stream; variable
1609 std::shared_ptr<stream> cpu_stream; variable
1720 std::shared_ptr<stream> cpu_stream; variable