Home
last modified time | relevance | path

Searched refs:rz_scratch_allocator (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc438 se::RedzoneAllocator rz_scratch_allocator( in operator ()() local
443 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator) in operator ()()
474 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones() in operator ()()
479 CheckRedzones(rz_scratch_allocator, &result); in operator ()()
Dconv_ops_3d.cc555 se::RedzoneAllocator rz_scratch_allocator( in launch() local
560 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator) in launch()
592 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones() in launch()
596 CheckRedzones(rz_scratch_allocator, &result); in launch()
Dconv_ops_fused_impl.h382 se::RedzoneAllocator rz_scratch_allocator(
387 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator)
408 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones()
412 CheckRedzones(rz_scratch_allocator, &result);
Dconv_grad_filter_ops.cc1043 se::RedzoneAllocator rz_scratch_allocator( in operator ()() local
1048 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator) in operator ()()
1080 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones() in operator ()()
1085 CheckRedzones(rz_scratch_allocator, &result); in operator ()()
Dconv_ops.cc1161 se::RedzoneAllocator rz_scratch_allocator( in operator ()() local
1167 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator) in operator ()()
1199 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones() in operator ()()
1204 CheckRedzones(rz_scratch_allocator, &result); in operator ()()
Dconv_grad_ops_3d.cc1564 se::RedzoneAllocator rz_scratch_allocator( in Compute() local
1569 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator) in Compute()
1601 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones() in Compute()
1607 CheckRedzones(rz_scratch_allocator, &result); in Compute()