Searched defs:net_args (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | mkl_util.h | 134 std::vector<std::unordered_map<int, memory>>& net_args) { in execute_primitives()
652 std::vector<MemoryArgsMap> net_args; in ConvertMklToTF() local
1272 std::vector<MemoryArgsMap> net_args; variable
1517 std::vector<MemoryArgsMap>& net_args, in CheckReorderToOpMem()
1552 std::vector<MemoryArgsMap> net_args; variable
1578 std::vector<MemoryArgsMap>& net_args, in CheckReorderToOpMem()
1619 std::vector<MemoryArgsMap> net_args; variable
1645 std::vector<MemoryArgsMap>& net_args, in CheckReorderToOpMem()
1697 std::vector<MemoryArgsMap>& net_args) { in InsertReorderToUserMem()
1717 std::vector<MemoryArgsMap> net_args; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_input_conversion_op.cc | 132 std::vector<MemoryArgsMap> net_args; in Compute() local
234 std::vector<MemoryArgsMap> net_args; in Compute() local
|
| H A D | mkl_lrn_op.cc | 173 std::vector<std::unordered_map<int, memory>> net_args; in Compute() local
436 std::vector<std::unordered_map<int, memory>> net_args; in Compute() local
|
| H A D | mkl_transpose_op.cc | 92 std::vector<MemoryArgsMap> net_args; in MKLTransposeND() local
|
| H A D | mkl_matmul_ops_common.h | 191 std::vector<std::unordered_map<int, memory>> net_args; member
694 std::vector<std::unordered_map<int, memory>> net_args; member
|
| H A D | mkl_concat_op.cc | 373 std::unordered_map<int, memory> net_args = { in Setup() local
756 std::unordered_map<int, memory> net_args = { in Compute() local
|
| H A D | mkl_pooling_ops_common.h | 136 std::vector<std::unordered_map<int, memory>> net_args; member
283 std::vector<std::unordered_map<int, memory>> net_args; member
|
| H A D | mkl_aggregate_ops.cc | 240 std::unordered_map<int, memory> net_args = { in Compute() local
|
| H A D | mkl_eltwise_activation_base_op.h | 99 std::vector<MemoryArgsMap> net_args; in Execute() local
|
| H A D | mkl_fused_batch_norm_op.cc | 181 std::vector<std::unordered_map<int, memory>> net_args; member
526 std::vector<std::unordered_map<int, memory>> net_args; member
|