Home
last modified time | relevance | path

Searched refs:TotalByteSize (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h89 int64 TotalByteSize() { return total_byte_size_; } in TotalByteSize() function
Dconv_grad_input_ops.cc433 : scratch_allocator.TotalByteSize()); in operator ()()
485 result.set_scratch_bytes(scratch_allocator.TotalByteSize()); in operator ()()
Dconv_ops_3d.cc555 : scratch_allocator.TotalByteSize()); in launch()
607 result.set_scratch_bytes(scratch_allocator.TotalByteSize()); in launch()
Dconv_grad_ops_3d.cc1469 : scratch_allocator.TotalByteSize()); in Compute()
1504 result.set_scratch_bytes(scratch_allocator.TotalByteSize()); in Compute()
1949 result.set_scratch_bytes(scratch_allocator.TotalByteSize()); in Compute()
1981 result.set_scratch_bytes(scratch_allocator.TotalByteSize()); in Compute()
Dconv_grad_filter_ops.cc1029 : scratch_allocator.TotalByteSize()); in operator ()()
1083 result.set_scratch_bytes(scratch_allocator.TotalByteSize()); in operator ()()
Dconv_ops.cc1041 : scratch_allocator.TotalByteSize()); in operator ()()
1095 result.set_scratch_bytes(scratch_allocator.TotalByteSize()); in operator ()()
Dcudnn_rnn_ops.cc389 int64 TotalByteSize() const { return total_byte_size_; } in TotalByteSize() function in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInTemp
432 int64 TotalByteSize() { return total_byte_size_; } in TotalByteSize() function in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInOutput
468 int64 TotalByteSize() { return total_byte_size_; } in TotalByteSize() function in tensorflow::__anonebbbcdba0111::CudnnRNNPersistentSpaceAllocator
Dfft_ops.cc390 int64 TotalByteSize() { return total_byte_size_; } in TotalByteSize() function in tensorflow::__anonbbe6ef150111::CufftScratchAllocator
Dconv_ops_fused_impl.h418 : scratch_allocator.TotalByteSize());
Dfused_batch_norm_op.cc694 int64 TotalByteSize() const { return total_byte_size_; } in TotalByteSize() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp
748 int64 TotalByteSize() { return total_byte_size_; } in TotalByteSize() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput