Home
last modified time | relevance | path

Searched refs:bwd_primitives (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_pooling_ops_common.cc184 context_.bwd_primitives.push_back(*context_.bwd); in Setup()
210 execute_primitives(context_.bwd_primitives, bwd_stream, context_.net_args); in Execute()
Dmkl_relu_op.cc272 DCHECK_EQ(context_.bwd_primitives.size(), in Execute()
274 execute_primitives(context_.bwd_primitives, bwd_stream, in Execute()
315 std::vector<mkldnn::primitive> bwd_primitives; member
369 context_.bwd_primitives.push_back(*context_.eltwise_bwd); in Setup()
Dmkl_fused_batch_norm_op.cc465 DCHECK_EQ(context_.bwd_primitives.size(), context_.net_args.size()); in Execute()
466 execute_primitives(context_.bwd_primitives, bwd_stream, context_.net_args); in Execute()
505 std::vector<mkldnn::primitive> bwd_primitives; member
577 context_.bwd_primitives.push_back(*context_.bn_bwd); in Setup()
Dmkl_pooling_ops_common.h276 std::vector<mkldnn::primitive> bwd_primitives; member