Searched defs:AssignStreams (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ | ||
| H A D | gpu_stream_util.cc | 30 Status AssignStreams(const Graph* graph, const AssignStreamsOpts& opts, function in namespace:tensorflow::gpu_stream_util 32 VLOG(1) << "AssignStreams"; |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ | ||
| H A D | stream_assignment.cc | 115 std::unique_ptr<StreamAssignment> AssignStreams(const HloModule& module) { function in namespace:xla::gpu |
Completed in 190 milliseconds