/external/tensorflow/tensorflow/python/ops/ |
D | collective_ops.py | 26 instance_key, argument 77 instance_key, argument 126 instance_key, argument 165 instance_key, argument 208 instance_key, argument 267 instance_key, argument 302 instance_key, argument 345 instance_key, argument
|
D | collective_ops_gpu_test.py | 254 def run_all_reduce(group_key, instance_key, merge_op): argument
|
D | collective_ops_test.py | 55 instance_key=1, argument
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | collective_param_resolver_local_test.cc | 211 void InitializeCollectiveParamsForBroadcast(int instance_key, int device_idx, in InitializeCollectiveParamsForBroadcast() 303 CollectiveParams* MakeCollectiveParams(int group_key, int instance_key, in MakeCollectiveParams() 348 int instance_key = 100; in TEST_F() local 397 int instance_key = 100; in TEST_F() local 422 auto complete_params = [this, &cancel_mgr](int group_key, int instance_key) { in TEST_F()
|
D | graph_execution_state.cc | 911 int32 instance_key; in BuildGraph() local 925 int32 instance_key; in BuildGraph() local 938 for (int32 instance_key : instance_key_set) { in BuildGraph() local
|
D | direct_session_test.cc | 2738 int instance_key) { in CollectiveFunction()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | collective_ops.cc | 29 int32 instance_key) { in CollectiveKey() 511 const Tensor& instance_key = c->input(3); in ComputeAsync() local 640 const Tensor& instance_key = c->input(3); in ComputeAsync() local 774 const Tensor& instance_key = c->input(3); in ComputeAsync() local 899 const Tensor& instance_key = c->input(2); in ComputeAsync() local
|
D | collective_nccl_test.cc | 110 void Init(const int num_ranks, const int instance_key) { in Init() 185 void RunTest(int num_ranks, int input_length, int instance_key) { in RunTest()
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | collective_ops_test.py | 56 def all_reduce(t, group_size, group_key, instance_key, *args, **kwargs): argument 64 def all_gather(t, group_size, group_key, instance_key, *args, **kwargs): argument 72 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key, argument 81 def broadcast_recv(shape, dtype, group_size, group_key, instance_key, *args, argument
|
/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/protobuf/ |
D | worker.proto | 577 int32 instance_key = 7; field 587 int32 instance_key = 1; field
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | collective_param_resolver_distributed.cc | 331 int32 instance_key) { in InstanceIsCached()
|
/external/tensorflow/tensorflow/core/framework/ |
D | collective.h | 108 int32 instance_key = -1; member
|