| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | collective_ops.py | 22 instance_key, argument
98 instance_key, argument
162 instance_key, argument
202 instance_key, argument
253 instance_key, argument
315 instance_key, argument
350 instance_key, argument
394 instance_key, argument
|
| H A D | collective_ops_gpu_test.py | 250 def run_all_reduce(group_key, instance_key, merge_op): argument
|
| H A D | collective_ops_test.py | 51 instance_key=1, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A 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()
|
| H A D | graph_execution_state.cc | 917 int32_t instance_key; in BuildGraph() local
931 int32_t instance_key; in BuildGraph() local
944 for (int32_t instance_key : instance_key_set) { in BuildGraph() local
|
| H A D | direct_session_test.cc | 2771 int instance_key) { in CollectiveFunction()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | collective_ops_test.py | 47 def all_reduce(t, group_size, group_key, instance_key, *args, **kwargs): argument
53 def all_gather(t, group_size, group_key, instance_key, *args, **kwargs): argument
65 def all_reduce(t, group_size, group_key, instance_key, *args, **kwargs): argument
73 def all_gather(t, group_size, group_key, instance_key, *args, **kwargs): argument
81 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key, argument
90 def broadcast_recv(shape, dtype, group_size, group_key, instance_key, *args, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | collective_ops.cc | 42 int32_t instance_key) { in CollectiveKey()
591 const Tensor& instance_key) { in FillCollectiveParams()
1081 int32 instance_key = group_id << 21 | resource->get_next_instance_key(); in FillCollectiveParams() local
|
| H A D | collective_nccl_test.cc | 111 void Init(const int num_ranks, const int instance_key) { in Init()
187 void RunTest(int num_ranks, int input_length, int instance_key) { in RunTest()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
| H A D | worker.proto | 579 int32 instance_key = 7; field
589 int32 instance_key = 1; field
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/ |
| H A D | import.cpp | 508 auto instance_key = std::rand(); in _load_for_mobile_impl() local
684 auto instance_key = std::rand(); in _load_extra_only_for_mobile() local
|
| H A D | module.cpp | 243 auto instance_key = std::rand(); in run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | collective_order.cc | 39 int32_t instance_key; in DiscoverDataDependencies() local
|
| H A D | collective_order_test.cc | 81 int instance_key) { in CollectiveReduceNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | collective_param_resolver_distributed.cc | 317 int32_t group_key, int32_t instance_key) { in InstanceIsCached()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | collective.h | 124 int32 instance_key = -1; member
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/ |
| D | worker.pb.h | 10519 inline ::int32_t CompleteInstanceRequest::instance_key() const { in instance_key() function
10700 inline ::int32_t CompleteInstanceResponse::instance_key() const { in instance_key() function
|