HomeSort by relevance Sort by last modified time
    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