Searched refs:fwd_primitives (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_softmax_op.cc | 72 DCHECK_EQ(context_.fwd_primitives.size(), context_.fwd_net_args.size()); in Execute()
73 execute_primitives(context_.fwd_primitives, fwd_cpu_stream, in Execute()
101 std::vector<mkldnn::primitive> fwd_primitives; member
137 context_.fwd_primitives.push_back(*context_.softmax_fwd); in Setup()
|
| D | mkl_pooling_ops_common.cc | 82 context_.fwd_primitives.push_back(*context_.fwd); in Setup()
110 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
|
| D | mkl_matmul_ops_common.h | 123 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
158 std::vector<mkldnn::primitive> fwd_primitives; member
283 context_.fwd_primitives.push_back(*context_.matmul_fwd); in Setup()
|
| D | mkl_relu_op.cc | 87 DCHECK_EQ(context_.fwd_primitives.size(), in Execute()
89 execute_primitives(context_.fwd_primitives, fwd_stream, in Execute()
120 std::vector<mkldnn::primitive> fwd_primitives; member
158 context_.fwd_primitives.push_back(*context_.eltwise_fwd); in Setup()
|
| D | mkl_concat_op.cc | 301 execute_primitives(context_.fwd_primitives, fwd_stream,
331 std::vector<mkldnn::primitive> fwd_primitives; member
373 context_.fwd_primitives.push_back(*context_.concat_fwd); in Setup()
|
| D | mkl_conv_ops.cc | 142 DCHECK_EQ(context_.fwd_primitives.size(), in Execute()
144 for (size_t i = 0; i < context_.fwd_primitives.size(); ++i) { in Execute()
145 context_.fwd_primitives.at(i).execute(*fwd_stream, in Execute()
193 std::vector<mkldnn::primitive> fwd_primitives; member
311 context_.fwd_primitives.push_back(*context_.conv_fwd); in Setup()
|
| D | mkl_fused_batch_norm_op.cc | 122 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
169 std::vector<mkldnn::primitive> fwd_primitives; member
319 context_.fwd_primitives.push_back(*context_.bn_fwd); in Setup()
|
| D | mkl_pooling_ops_common.h | 132 std::vector<mkldnn::primitive> fwd_primitives; member
|