Home
last modified time | relevance | path

Searched refs:TotalAllocatedBytesExcludingRedzones (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dredzone_allocator.h55 int64 TotalAllocatedBytesExcludingRedzones() const { in TotalAllocatedBytesExcludingRedzones() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc491 scratch_allocator.TotalAllocatedBytesExcludingRedzones(); in PickBestAlgorithmNoCacheCuda()
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc432 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones() in operator ()()
Dconv_ops_3d.cc554 .TotalAllocatedBytesExcludingRedzones() in launch()
Dconv_ops_fused_impl.h417 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones()
Dconv_grad_filter_ops.cc1028 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones() in operator ()()
Dconv_ops.cc1040 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones() in operator ()()
Dconv_grad_ops_3d.cc1468 .TotalAllocatedBytesExcludingRedzones() in Compute()