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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
scratch_allocator.cc 24 if (byte_size > GetMemoryLimitInBytes(stream)) {
29 byte_size, GetMemoryLimitInBytes(stream)));
scratch_allocator.h 35 int64 GetMemoryLimitInBytes(se::Stream* stream) override {
fft_thunk.h 43 int64 GetMemoryLimitInBytes(se::Stream* stream) override;
fft_thunk.cc 35 int64 FftScratchAllocator::GetMemoryLimitInBytes(se::Stream* stream) {
43 if (byte_size > GetMemoryLimitInBytes(stream)) {
48 byte_size, GetMemoryLimitInBytes(stream)));
cudnn_conv_runner.cc 87 int64 GetMemoryLimitInBytes(se::Stream* /*stream*/) override {
  /external/tensorflow/tensorflow/stream_executor/
scratch_allocator.h 48 virtual int64 GetMemoryLimitInBytes(Stream* stream) = 0;
69 int64 GetMemoryLimitInBytes(Stream* stream) override;
scratch_allocator.cc 28 int64 OneTimeScratchAllocator::GetMemoryLimitInBytes(Stream* stream) {
  /external/tensorflow/tensorflow/core/kernels/
conv_ops_gpu.h 44 int64 GetMemoryLimitInBytes(se::Stream* stream) override {
fft_ops.cc 316 int64 GetMemoryLimitInBytes(se::Stream* stream) override {
cudnn_rnn_ops.cc 366 int64 GetMemoryLimitInBytes(Stream* stream) override {
412 int64 GetMemoryLimitInBytes(Stream* stream) override {
452 int64 GetMemoryLimitInBytes(Stream* stream) override {
    [all...]
batch_matmul_op_impl.h 263 int64 GetMemoryLimitInBytes(Stream* stream) override { return -1; }
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.cc     [all...]

Completed in 1347 milliseconds