HomeSort by relevance Sort by last modified time
    Searched refs:InfeedManager (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
infeed_manager.cc 23 se::Stream* InfeedManager::GetStream(se::StreamExecutor* executor) {
40 InfeedManager* GetOrCreateInfeedManager() {
41 static InfeedManager* manager = new InfeedManager;
infeed_manager.h 64 class InfeedManager : public XfeedQueue<ShapeTree<InfeedBuffer>> {
85 InfeedManager* GetOrCreateInfeedManager();
gpu_transfer_manager.cc 76 gpu::InfeedManager* infeed_manager = gpu::GetOrCreateInfeedManager();
106 gpu::InfeedManager* infeed_manager = gpu::GetOrCreateInfeedManager();

Completed in 106 milliseconds