HomeSort by relevance Sort by last modified time
    Searched refs:fwd_stream (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_fused_batch_norm_op.cc 51 context_.fwd_stream.reset(new mkldnn::stream(mkldnn::stream::kind::eager));
80 context_.fwd_stream->submit(context_.fwd_primitives);
125 std::shared_ptr<mkldnn::stream> fwd_stream; member in struct:tensorflow::MklFusedBatchNormFwdPrimitive::BatchNormFwdContext
137 fwd_stream(nullptr) {}
    [all...]
mkl_relu_op.cc 66 context_.fwd_stream.reset(new stream(stream::kind::eager));
83 context_.fwd_stream->submit(context_.fwd_primitives);
120 std::shared_ptr<stream> fwd_stream; member in struct:tensorflow::MklEltwiseFwdPrimitive::EltwiseFwdContext
133 fwd_stream(nullptr) {}
    [all...]
mkl_pooling_ops_common.h 74 context_.fwd_stream.reset(new stream(stream::kind::eager));
130 std::shared_ptr<mkldnn::stream> fwd_stream; member in struct:tensorflow::MklPoolingFwdPrimitive::PoolingFwdContext
145 fwd_stream(nullptr) {}
mkl_conv_ops.cc 106 context_.fwd_stream.reset(new stream(stream::kind::eager));
130 context_.fwd_stream->submit(context_.fwd_primitives);
153 context_.fwd_stream->submit(context_.fwd_primitives);
195 std::shared_ptr<mkldnn::stream> fwd_stream; member in struct:tensorflow::MklConvFwdPrimitive::ConvFwdContext
211 fwd_stream(nullptr) {}
    [all...]
mkl_pooling_ops_common.cc 116 context_.fwd_stream->submit(context_.fwd_primitives);

Completed in 311 milliseconds