Searched defs:to_device_ctx (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | collective_rma_distributed_test.cc | 391 DeviceContext* to_device_ctx = nullptr; in TEST_P() local
423 DeviceContext* to_device_ctx = nullptr; in TEST_P() local
459 DeviceContext* to_device_ctx = nullptr; in TEST_P() local
495 DeviceContext* to_device_ctx = nullptr; in TEST_P() local
528 DeviceContext* to_device_ctx = nullptr; in TEST_P() local
|
| H A D | collective_rma_distributed.cc | 43 DeviceContext* to_device_ctx, in RecvBufCall()
110 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | collective_rma_local.cc | 28 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|
| H A D | collective_test_util.cc | 60 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|