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

  /external/tensorflow/tensorflow/core/kernels/
mkl_fused_batch_norm_op.cc 292 context_.bwd_stream.reset(new mkldnn::stream(mkldnn::stream::kind::eager));
328 context_.bwd_stream->submit(context_.bwd_primitives);
370 std::shared_ptr<mkldnn::stream> bwd_stream; member in struct:tensorflow::MklFusedBatchNormBwdPrimitive::BatchNormBwdContext
380 bwd_stream(nullptr) {}
    [all...]
mkl_relu_op.cc 251 context_.bwd_stream.reset(new stream(stream::kind::eager));
270 context_.bwd_stream->submit(context_.bwd_primitives);
318 std::shared_ptr<stream> bwd_stream; member in struct:tensorflow::MklEltwiseBwdPrimitive::EltwiseBwdContext
336 bwd_stream(nullptr) {}
    [all...]
mkl_pooling_ops_common.h 215 context_.bwd_stream.reset(new stream(stream::kind::eager));
279 std::shared_ptr<mkldnn::stream> bwd_stream; member in struct:tensorflow::MklPoolingBwdPrimitive::PoolingBwdContext
297 bwd_stream(nullptr) {}
mkl_pooling_ops_common.cc 209 context_.bwd_stream->submit(context_.bwd_primitives);

Completed in 430 milliseconds