Searched defs:net_args (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 144 std::vector<MemoryArgsMap> net_args; in Compute() local
246 std::vector<MemoryArgsMap> net_args; in Compute() local
|
| D | mkl_transpose_op.cc | 151 std::vector<MemoryArgsMap> net_args; in MKLTransposeND() local
|
| D | mkl_aggregate_ops.cc | 240 std::unordered_map<int, memory> net_args = { in Compute() local
|
| D | mkl_lrn_op.cc | 172 std::vector<std::unordered_map<int, memory>> net_args; in Compute() local
435 std::vector<std::unordered_map<int, memory>> net_args; in Compute() local
|
| D | mkl_matmul_ops_common.h | 160 std::vector<std::unordered_map<int, memory>> net_args; member
582 std::vector<std::unordered_map<int, memory>> net_args; member
|
| D | mkl_concat_op.cc | 366 std::unordered_map<int, memory> net_args = { in Setup() local
743 std::unordered_map<int, memory> net_args = { in Compute() local
|
| 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_fused_batch_norm_op.cc | 171 std::vector<std::unordered_map<int, memory>> net_args; member
507 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
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 132 std::vector<std::unordered_map<int, memory>>& net_args) { in execute_primitives()
649 std::vector<MemoryArgsMap> net_args; in ConvertMklToTF() local
1235 std::vector<MemoryArgsMap> net_args; variable
1472 std::vector<MemoryArgsMap>& net_args, in CheckReorderToOpMem()
1501 std::vector<MemoryArgsMap> net_args; variable
1527 std::vector<MemoryArgsMap>& net_args, in CheckReorderToOpMem()
1562 std::vector<MemoryArgsMap> net_args; variable
1588 std::vector<MemoryArgsMap>& net_args, in CheckReorderToOpMem()
1641 std::vector<MemoryArgsMap>& net_args) { in InsertReorderToUserMem()
1661 std::vector<MemoryArgsMap> net_args; variable
|
| D | mkl_types.h | 34 #define GET_CHECK_REORDER_MEM_ARGS(md, tensor, net, net_args, engine) \ argument
36 #define GET_CHECK_REORDER_TO_OP_MEM_ARGS(md, tensor, net, net_args, engine) \ argument
|