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.cc412 se::RedzoneAllocator rz_scratch_allocator( in operator ()() local
417 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator) in operator ()()
432 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones() in operator ()()
437 CheckRedzones(rz_scratch_allocator, &result); in operator ()()
Dconv_ops_3d.cc532 se::RedzoneAllocator rz_scratch_allocator( in launch() local
537 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator) in launch()
553 ? rz_scratch_allocator in launch()
558 CheckRedzones(rz_scratch_allocator, &result); in launch()
Dconv_ops_fused_impl.h396 se::RedzoneAllocator rz_scratch_allocator(
401 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator)
417 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones()
421 CheckRedzones(rz_scratch_allocator, &result);
Dconv_grad_filter_ops.cc1006 se::RedzoneAllocator rz_scratch_allocator( in operator ()() local
1011 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator) in operator ()()
1028 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones() in operator ()()
1033 CheckRedzones(rz_scratch_allocator, &result); in operator ()()
Dconv_ops.cc1017 se::RedzoneAllocator rz_scratch_allocator( in operator ()() local
1023 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator) in operator ()()
1040 ? rz_scratch_allocator.TotalAllocatedBytesExcludingRedzones() in operator ()()
1045 CheckRedzones(rz_scratch_allocator, &result); in operator ()()
Dconv_grad_ops_3d.cc1446 se::RedzoneAllocator rz_scratch_allocator( in Compute() local
1451 ? static_cast<se::ScratchAllocator*>(&rz_scratch_allocator) in Compute()
1467 ? rz_scratch_allocator in Compute()
1474 CheckRedzones(rz_scratch_allocator, &result); in Compute()