Searched defs:exec_key (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | collective_nccl_test.cc | 328 string exec_key = in RunReduce() local
360 string exec_key = in RunBroadcast() local
401 string exec_key = in RunGather() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | collective.h | 317 const string& exec_key, StatusCallback done) { in ExecuteAsync()
398 const string exec_key; member
|
| H A D | collective.cc | 159 const string& exec_key, int64_t step_id, const Tensor* input, in CollectiveContext()
|
| /aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/_utilities/ |
| H A D | profiling_helper.py | 90 def __get_backend(exec_key): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/nccl/ |
| H A D | collective_communicator.cc | 65 string NcclCollectiveKey(const string& exec_key, int step_id) { in NcclCollectiveKey()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | ring_alg.cc | 64 string RingAlgBufKey(const string& name, const string& exec_key, int pass, in RingAlgBufKey()
|
| H A D | hierarchical_tree_broadcaster.cc | 45 string BroadcastBufKey(const string& exec_key, int subdiv, int src_rank, in BroadcastBufKey()
|
| H A D | base_collective_executor.cc | 274 const string& exec_key, in ExecuteAsync()
|
| H A D | collective_test_util.cc | 351 string exec_key = strings::StrCat(col_params->instance.instance_key, ":0:0"); in RunCollective() local
|