OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:BorrowStream
(Results
1 - 8
of
8
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
backend.h
66
// StreamPtr stream = backend->
BorrowStream
().ConsumeValueOrDie();
119
StatusOr<StreamPtr>
BorrowStream
(int device_ordinal);
120
StatusOr<StreamPtr>
BorrowStream
(
123
// Returns a function to borrow a stream, as `
BorrowStream
` above does.
127
return [this](int device_ordinal) { return
BorrowStream
(device_ordinal); };
179
// Mapping from stream executor to stream pools, used by `
BorrowStream
` above.
service_executable_run_options.h
54
StatusOr<Pool<perftools::gputools::Stream>::SmartPtr>
BorrowStream
(
backend.cc
103
StatusOr<Backend::StreamPtr> Backend::
BorrowStream
(int device_ordinal) {
105
return
BorrowStream
(exec);
108
StatusOr<Backend::StreamPtr> Backend::
BorrowStream
(
service.cc
518
backend->
BorrowStream
(replicas[replica]));
641
backend->
BorrowStream
(executor));
[
all
...]
/external/tensorflow/tensorflow/compiler/jit/
xla_device.cc
218
TF_ASSIGN_OR_RETURN(stream_, backend->
BorrowStream
(device_ordinal_));
/external/tensorflow/tensorflow/compiler/xla/client/
local_client.cc
39
return backend->
BorrowStream
(device_ordinal);
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_executable.cc
161
run_options->
BorrowStream
(main_stream->parent()->device_ordinal()));
/external/tensorflow/tensorflow/compiler/xla/tests/
xla_hlo_profile_test.cc
156
backend->
BorrowStream
(backend->default_device_ordinal()));
Completed in 629 milliseconds