Home
last modified time | relevance | path

Searched defs:instance_key (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcollective_ops.py26 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
Dcollective_ops_gpu_test.py254 def run_all_reduce(group_key, instance_key, merge_op): argument
Dcollective_ops_test.py55 instance_key=1, argument
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local_test.cc211 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()
Dgraph_execution_state.cc911 int32 instance_key; in BuildGraph() local
925 int32 instance_key; in BuildGraph() local
938 for (int32 instance_key : instance_key_set) { in BuildGraph() local
Ddirect_session_test.cc2738 int instance_key) { in CollectiveFunction()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc29 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
Dcollective_nccl_test.cc110 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/
Dcollective_ops_test.py56 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/
Dcollective_order.cc39 int32 instance_key; in DiscoverDataDependencies() local
Dcollective_order_test.cc81 int instance_key) { in CollectiveReduceNode()
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto577 int32 instance_key = 7; field
587 int32 instance_key = 1; field
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc331 int32 instance_key) { in InstanceIsCached()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h108 int32 instance_key = -1; member