| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 144 std::vector<MemoryArgsMap> net_args; in Compute() local
149 input.CheckReorderToOpMem(input1_md, tensor_out, net, net_args, in Compute()
153 ExecutePrimitive(net, &net_args, cpu_engine, context); in Compute()
246 std::vector<MemoryArgsMap> net_args; in Compute() local
248 net, net_args, cpu_engine); in Compute()
260 ExecutePrimitive(net, &net_args, cpu_engine, context); in Compute()
|
| D | mkl_pooling_ops_common.cc | 72 context_.net_args.push_back({{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
77 context_.net_args.push_back({{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
110 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
173 context_.net_args.push_back( in Setup()
179 context_.net_args.push_back( in Setup()
210 execute_primitives(context_.bwd_primitives, bwd_stream, context_.net_args); in Execute()
|
| D | mkl_fused_batch_norm_op.cc | 122 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
171 std::vector<std::unordered_map<int, memory>> net_args; member
238 context_.net_args.push_back( in Setup()
243 context_.net_args.push_back({{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
250 context_.net_args.push_back( in Setup()
258 context_.net_args.push_back( in Setup()
267 context_.net_args.push_back( in Setup()
274 context_.net_args.push_back( in Setup()
285 context_.net_args.push_back( in Setup()
293 context_.net_args.push_back( in Setup()
[all …]
|
| D | mkl_transpose_op.cc | 151 std::vector<MemoryArgsMap> net_args; in MKLTransposeND() local
152 net_args.push_back({{MKLDNN_ARG_FROM, *in.GetUsrMem()}, in MKLTransposeND()
154 execute_primitives(net, transpose_stream, net_args); in MKLTransposeND()
|
| D | mkl_aggregate_ops.cc | 240 std::unordered_map<int, memory> net_args = { in Compute() local
243 net_args.insert({MKLDNN_ARG_MULTIPLE_SRC + i, inputs[i]}); in Compute()
245 sum_op.execute(*fwd_cpu_stream, net_args); in Compute()
|
| D | mkl_lrn_op.cc | 172 std::vector<std::unordered_map<int, memory>> net_args; in Compute() local
173 net_args.push_back({{MKLDNN_ARG_SRC, src_dnn_data.GetOpMem()}, in Compute()
177 net.at(0).execute(*fwd_stream_, net_args.at(0)); in Compute()
435 std::vector<std::unordered_map<int, memory>> net_args; in Compute() local
437 net_args.push_back({{MKLDNN_ARG_SRC, orig_input_dnn_data.GetOpMem()}, in Compute()
441 net.at(0).execute(*bwd_stream_, net_args.at(0)); in Compute()
|
| D | mkl_matmul_ops_common.h | 123 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
160 std::vector<std::unordered_map<int, memory>> net_args; member
278 context_.net_args.push_back({{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
555 execute_primitives(context_.matmul_primitives, stream, context_.net_args); in Execute()
582 std::vector<std::unordered_map<int, memory>> net_args; member
624 context_.net_args.push_back({{MKLDNN_ARG_SRC, *context_.a_mem}, in Setup()
|
| D | mkl_concat_op.cc | 366 std::unordered_map<int, memory> net_args = { in Setup() local
369 net_args.insert({MKLDNN_ARG_MULTIPLE_SRC + i, context_.data_mem[i]}); in Setup()
372 context_.fwd_primitives_args.push_back(net_args); in Setup()
743 std::unordered_map<int, memory> net_args = { in Compute() local
746 net_args.insert({MKLDNN_ARG_MULTIPLE_SRC + i, inputs[i]}); in Compute()
748 concat_op.execute(*fwd_cpu_stream, net_args); in Compute()
|
| D | mkl_pooling_ops_common.h | 134 std::vector<std::unordered_map<int, memory>> net_args; member
277 std::vector<std::unordered_map<int, memory>> net_args; member
|
| D | mkl_conv_ops.cc | 1059 std::vector<std::unordered_map<int, memory>> net_args; in PrepareAndExecuteNet() local
1063 net_args.push_back({{MKLDNN_ARG_SRC, src->GetOpMem()}, in PrepareAndExecuteNet()
1070 net_args.push_back({{MKLDNN_ARG_SRC, src->GetOpMem()}, in PrepareAndExecuteNet()
1074 ExecutePrimitive(net, &net_args, cpu_engine_); in PrepareAndExecuteNet()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_types.h | 34 #define GET_CHECK_REORDER_MEM_ARGS(md, tensor, net, net_args, engine) \ argument
35 md, tensor, net, net_args, engine
36 #define GET_CHECK_REORDER_TO_OP_MEM_ARGS(md, tensor, net, net_args, engine) \ argument
37 md, tensor, net, net_args, engine
101 #define NET_ARGS_PTR &net_args
|
| D | mkl_util.h | 132 std::vector<std::unordered_map<int, memory>>& net_args) { in execute_primitives() argument
133 DCHECK_EQ(primitives.size(), net_args.size()); in execute_primitives()
135 primitives.at(i).execute(*stream, net_args.at(i)); in execute_primitives()
610 const std::vector<MemoryArgsMap>* net_args,
613 DCHECK(net_args);
614 DCHECK_EQ(net.size(), net_args->size());
617 net.at(i).execute(*cpu_stream, net_args->at(i));
649 std::vector<MemoryArgsMap> net_args; in ConvertMklToTF() local
651 net, net_args, cpu_engine); in ConvertMklToTF()
656 ExecutePrimitive(net, &net_args, cpu_engine, context); in ConvertMklToTF()
[all …]
|