Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_pooling_ops_common.cc101 context_.fwd_primitives.push_back(*context_.fwd); in Setup()
116 context_.fwd_stream->submit(context_.fwd_primitives); in Execute()
Dmkl_fused_batch_norm_op.cc80 context_.fwd_stream->submit(context_.fwd_primitives); in Execute()
126 std::vector<mkldnn::primitive> fwd_primitives; member
211 context_.fwd_primitives.push_back(*context_.bn_fwd); in Setup()
Dmkl_relu_op.cc83 context_.fwd_stream->submit(context_.fwd_primitives); in Execute()
121 std::vector<mkldnn::primitive> fwd_primitives; member
159 context_.fwd_primitives.push_back(*context_.eltwise_fwd); in Setup()
Dmkl_conv_ops.cc130 context_.fwd_stream->submit(context_.fwd_primitives); in Execute()
153 context_.fwd_stream->submit(context_.fwd_primitives); in Execute()
196 std::vector<mkldnn::primitive> fwd_primitives; member
311 context_.fwd_primitives.push_back(*context_.conv_fwd); in Setup()
Dmkl_pooling_ops_common.h131 std::vector<mkldnn::primitive> fwd_primitives; member