Home
last modified time | relevance | path

Searched defs:TotalByteSize (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_gpu.h85 int64 TotalByteSize() { return total_byte_size_; } in TotalByteSize() function
H A Dfused_batch_norm_op.cc704 int64_t TotalByteSize() const { return total_byte_size_; } in TotalByteSize() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp
758 int64_t TotalByteSize() { return total_byte_size_; } in TotalByteSize() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput
H A Dcudnn_rnn_ops.cc392 int64_t TotalByteSize() const { return total_byte_size_; } in TotalByteSize() function in tensorflow::__anon980e2df00111::CudnnRnnAllocatorInTemp
435 int64_t TotalByteSize() { return total_byte_size_; } in TotalByteSize() function in tensorflow::__anon980e2df00111::CudnnRnnAllocatorInOutput
471 int64_t TotalByteSize() { return total_byte_size_; } in TotalByteSize() function in tensorflow::__anon980e2df00111::CudnnRNNSpaceAllocator
H A Dfft_ops.cc401 int64_t TotalByteSize() { return total_byte_size_; } in TotalByteSize() function in tensorflow::__anon055adecb0111::CufftScratchAllocator