Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A 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
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/
H A DLlgaTensorImpl.cpp42 static dnnl::stream cpu_stream{Engine::getEngine()}; in getStream() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_batch_matmul_op.cc161 std::shared_ptr<stream> cpu_stream; in Compute() local
H A Dmkl_matmul_op_fused.cc265 std::shared_ptr<stream> cpu_stream; in Compute() local
H A Dmkl_einsum_op.cc121 std::shared_ptr<stream> cpu_stream; in MKLContractOperands() local
H A Dmkl_qmatmul_op.cc293 std::shared_ptr<stream> cpu_stream; in Compute() local
H A Dmkl_quantize_op.cc485 std::shared_ptr<stream> cpu_stream; in Compute() local
H A Dmkl_matmul_ops_common.h915 std::shared_ptr<stream> cpu_stream; variable