Home
last modified time | relevance | path

Searched refs:MemoryArgsMap (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_util.h52 using MemoryArgsMap = std::unordered_map<int, memory>; variable
607 const std::vector<MemoryArgsMap>* net_args,
652 std::vector<MemoryArgsMap> net_args; in ConvertMklToTF()
1272 std::vector<MemoryArgsMap> net_args;
1517 std::vector<MemoryArgsMap>& net_args, in CheckReorderToOpMem()
1525 net_args.push_back(MemoryArgsMap{{DNNL_ARG_FROM, *user_memory_}, in CheckReorderToOpMem()
1552 std::vector<MemoryArgsMap> net_args;
1578 std::vector<MemoryArgsMap>& net_args, in CheckReorderToOpMem()
1586 net_args.push_back(MemoryArgsMap{{DNNL_ARG_FROM, *user_memory_}, in CheckReorderToOpMem()
1619 std::vector<MemoryArgsMap> net_args;
[all …]
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_input_conversion_op.cc132 std::vector<MemoryArgsMap> net_args; in Compute()
234 std::vector<MemoryArgsMap> net_args; in Compute()
Dmkl_transpose_op.cc92 std::vector<MemoryArgsMap> net_args; in MKLTransposeND()
Dmkl_softmax_op.cc110 std::vector<MemoryArgsMap> fwd_net_args;
Dmkl_eltwise_activation_base_op.h99 std::vector<MemoryArgsMap> net_args; in Execute()
Dmkl_conv_grad_filter_ops.cc178 std::vector<MemoryArgsMap> bwd_filter_primitives_args;
Dmkl_relu_op.cc318 std::vector<MemoryArgsMap> bwd_primitives_args;