Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_softmax_op.cc65 std::shared_ptr<stream> fwd_cpu_stream) { in Execute()
323 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
H A Dmkl_concat_op.cc746 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
784 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
H A Dmkl_aggregate_ops.cc176 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
H A Dmkl_avgpooling_op.cc128 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
H A Dmkl_maxpooling_op.cc150 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
H A Dmkl_fused_batch_norm_op.cc929 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
H A Dmkl_relu_op.cc482 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local
H A Dmkl_conv_ops.cc889 std::shared_ptr<stream> fwd_cpu_stream; in Compute() local