Home
last modified time | relevance | path

Searched refs:infeed_manager (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_transfer_manager.cc76 gpu::InfeedManager* infeed_manager = gpu::GetOrCreateInfeedManager(); in EnqueueBuffersToInfeed() local
77 se::Stream* stream = infeed_manager->GetStream(executor); in EnqueueBuffersToInfeed()
89 infeed_manager->EnqueueDestination(std::move(buffers)); in EnqueueBuffersToInfeed()
106 gpu::InfeedManager* infeed_manager = gpu::GetOrCreateInfeedManager(); in TransferBufferToInfeedInternal() local
107 se::Stream* stream = infeed_manager->GetStream(executor); in TransferBufferToInfeedInternal()
DBUILD373 ":infeed_manager",
782 ":infeed_manager",
910 name = "infeed_manager",
911 srcs = ["infeed_manager.cc"],
912 hdrs = ["infeed_manager.h"],
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dkeras_support.py939 def _specialize_model(self, input_specs, infeed_manager): argument
1100 sized_infeed = infeed_manager.build_infeed_from_input_specs(
1159 def _tpu_model_ops_for_input_specs(self, input_specs, infeed_manager): argument
1171 if input_specs is None or infeed_manager is None:
1186 infeed_manager)
1271 infeed_manager = self._lookup_infeed_manager(inputs)
1273 infeed_instance = infeed_manager.make_infeed_instance(inputs)
1277 infeed_manager)
1515 infeed_manager = TPUDatasetInfeedManager(
1519 x = infeed_manager.dummy_x
[all …]
/external/tensorflow/tensorflow/compiler/xla/tools/
DBUILD70 "//tensorflow/compiler/xla/service/gpu:infeed_manager",