Home
last modified time | relevance | path

Searched defs:dev_ctx (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dbuf_rendezvous.cc66 DeviceContext* dev_ctx, const Tensor* v, in ProvideBuf()
Dring_gatherer_test.cc453 DeviceContext* dev_ctx = nullptr; in DoGather() local
Dring_reducer_test.cc482 DeviceContext* dev_ctx = nullptr; in DoReduce() local
Dhierarchical_tree_broadcaster_test.cc641 DeviceContext* dev_ctx = nullptr; in DoBroadcast() local
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer_test.cc260 DeviceContext* dev_ctx = nullptr; in DoReduce() local