Home
last modified time | relevance | path

Searched refs:fwd_stream (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_pooling_ops_common.h74 context_.fwd_stream.reset(new stream(stream::kind::eager)); in MklPoolingFwdPrimitive()
130 std::shared_ptr<mkldnn::stream> fwd_stream; member
145 fwd_stream(nullptr) {} in PoolingFwdContext()
Dmkl_fused_batch_norm_op.cc51 context_.fwd_stream.reset(new mkldnn::stream(mkldnn::stream::kind::eager)); in MklFusedBatchNormFwdPrimitive()
80 context_.fwd_stream->submit(context_.fwd_primitives); in Execute()
125 std::shared_ptr<mkldnn::stream> fwd_stream; member
137 fwd_stream(nullptr) {} in BatchNormFwdContext()
Dmkl_relu_op.cc66 context_.fwd_stream.reset(new stream(stream::kind::eager)); in MklEltwiseFwdPrimitive()
83 context_.fwd_stream->submit(context_.fwd_primitives); in Execute()
120 std::shared_ptr<stream> fwd_stream; member
133 fwd_stream(nullptr) {} in EltwiseFwdContext()
Dmkl_conv_ops.cc106 context_.fwd_stream.reset(new stream(stream::kind::eager)); in MklConvFwdPrimitive()
130 context_.fwd_stream->submit(context_.fwd_primitives); in Execute()
153 context_.fwd_stream->submit(context_.fwd_primitives); in Execute()
195 std::shared_ptr<mkldnn::stream> fwd_stream; member
211 fwd_stream(nullptr) {} in ConvFwdContext()
Dmkl_pooling_ops_common.cc116 context_.fwd_stream->submit(context_.fwd_primitives); in Execute()