| /external/armnn/python/pyarmnn/src/pyarmnn/_utilities/ |
| D | profiling_helper.py | 64 for exec_key, exec_value in execution_data.items():
70 time_data["backend"] = __get_backend(exec_key)
71 workload_data[exec_key] = time_data
73 if exec_key.startswith("Wall clock time_#") and exec_value["type"] == "Measurement":
90 def __get_backend(exec_key): argument
91 if "ref" in exec_key.lower():
93 elif "neon" in exec_key.lower():
95 elif "cl" in exec_key.lower():
97 elif "ethos" in exec_key.lower():
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 65 string NcclCollectiveKey(const string& exec_key, int step_id) { in NcclCollectiveKey() argument
66 return strings::StrCat(exec_key, ":", step_id); in NcclCollectiveKey()
88 NcclCollectiveKey(col_ctx->exec_key, col_ctx->step_id); in Enqueue()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.cc | 64 string RingAlgBufKey(const string& name, const string& exec_key, int pass, in RingAlgBufKey() argument
67 return strings::StrCat(name, "(", exec_key, "):pass(", pass, "):section(", in RingAlgBufKey()
72 return strings::StrCat(exec_key, ":", pass, ":", section, ":", source_rank); in RingAlgBufKey()
399 string send_buf_key = RingAlgBufKey(name_, col_ctx_->exec_key, in DispatchSend()
419 RingAlgBufKey(name_, col_ctx_->exec_key, rf->second_pass, rf->sc_idx, in DispatchRecv()
440 " exec ", col_ctx_->exec_key, " step_id=", col_ctx_->step_id, in FieldState()
|
| D | hierarchical_tree_broadcaster.cc | 45 string BroadcastBufKey(const string& exec_key, int subdiv, int src_rank, in BroadcastBufKey() argument
48 return strings::StrCat("broadcast(", exec_key, "):subdiv(", subdiv, in BroadcastBufKey()
52 return strings::StrCat(exec_key, ":", subdiv, ":", src_rank, ":", dst_rank); in BroadcastBufKey()
416 BroadcastBufKey(col_ctx_->exec_key, subdiv, src_rank, dst_rank); in DispatchSend()
437 BroadcastBufKey(col_ctx_->exec_key, subdiv, src_rank, dst_rank); in DispatchRecv()
|
| D | permuter.cc | 83 strings::StrCat(col_ctx_->exec_key, src_rank, target_rank); in DispatchSend()
99 strings::StrCat(col_ctx_->exec_key, src_rank, target_rank); in DispatchRecv()
|
| D | all_to_all.cc | 126 strings::StrCat(col_ctx_->exec_key, src_rank, target_rank); in DispatchSend()
138 strings::StrCat(col_ctx_->exec_key, src_rank, target_rank); in DispatchRecv()
|
| D | base_collective_executor.h | 113 const string& exec_key, StatusCallback done) override;
|
| D | collective_test_util.cc | 351 string exec_key = strings::StrCat(col_params->instance.instance_key, ":0:0"); in RunCollective() local
354 test_env->device_mgr.get(), &ctx, &op_params, col_params, exec_key, in RunCollective()
|
| D | base_collective_executor.cc | 274 const string& exec_key, in ExecuteAsync() argument
324 col_params, exec_key, step_id_, input, output); in ExecuteAsync()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 159 const string& exec_key, int64_t step_id, const Tensor* input, in CollectiveContext() argument
167 exec_key(exec_key), in CollectiveContext()
|
| D | collective.h | 317 const string& exec_key, StatusCallback done) { in ExecuteAsync() argument
398 const string exec_key; member
410 const CollectiveParams* col_params, const string& exec_key,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 328 string exec_key = in RunReduce() local
334 /*OpKernelContext=*/&ctx, &op_params, col_params_, exec_key, kStepId, in RunReduce()
360 string exec_key = in RunBroadcast() local
366 /*OpKernelContext=*/&ctx, &op_params, col_params_, exec_key, kStepId, in RunBroadcast()
401 string exec_key = in RunGather() local
407 /*OpKernelContext=*/&ctx, &op_params, col_params_, exec_key, kStepId, in RunGather()
|