Searched defs:instance_key (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 24 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op, argument
61 def all_gather(t, group_size, group_key, instance_key): argument
90 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key): argument
137 def broadcast_recv(shape, dtype, group_size, group_key, instance_key): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 208 int32 instance_key = col_params_.instance.instance_key; in ComputeAsync() local
277 int32 instance_key = col_params_.instance.instance_key; in ComputeAsync() local
339 int32 instance_key = col_params_.instance.instance_key; in ComputeAsync() local
|
| D | collective_nccl_reducer_test.cc | 121 const int instance_key = 23; in Init() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 799 int32 instance_key; in BuildGraph() local
810 int32 instance_key; in BuildGraph() local
821 for (int32 instance_key : instance_key_set) { in BuildGraph() local
|
| D | collective_param_resolver_local_test.cc | 203 void InitializeCollectiveParamsForBroadcast(int instance_key, int device_idx, in InitializeCollectiveParamsForBroadcast()
|
| D | direct_session_test.cc | 2279 int instance_key) { in CollectiveFunction()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 526 int32 instance_key = 7; field
536 int32 instance_key = 1; field
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | collective_order.cc | 39 int32 instance_key; in DiscoverDataDependencies() local
|
| D | collective_order_test.cc | 81 int instance_key) { in CollectiveReduceNode()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 284 bool CollectiveParamResolverDistributed::InstanceIsCached(int32 instance_key) { in InstanceIsCached()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 81 int32 instance_key = -1; member
|