Searched refs:DnnScratchAllocator (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 51 class DnnScratchAllocator : public se::ScratchAllocator {
53 virtual ~DnnScratchAllocator() {} in ~DnnScratchAllocator()
54 DnnScratchAllocator(int64 memory_limit, OpKernelContext* context) in DnnScratchAllocator() function
|
| D | cudnn_pooling_gpu.cc | 107 DnnScratchAllocator scratch_allocator(PoolingScratchSize, context); in Compute()
249 DnnScratchAllocator scratch_allocator(PoolingScratchSize, context); in Compute()
|
| D | conv_grad_input_ops.cc | 359 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, ctx); in operator ()()
410 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in operator ()()
442 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, ctx); in operator ()()
|
| D | conv_ops_3d.cc | 531 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()
564 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()
624 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()
|
| D | conv_grad_ops_3d.cc | 1444 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute()
1480 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute()
1521 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute()
1935 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in Compute()
1956 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in Compute()
2014 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in Compute()
|
| D | pooling_ops_common.cc | 395 DnnScratchAllocator scratch_allocator(PoolingScratchSize, context); in Compute()
713 DnnScratchAllocator scratch_allocator(PoolingScratchSize, context); in Compute()
|
| D | conv_grad_filter_ops.cc | 1004 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in operator ()()
1038 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in operator ()()
1099 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, ctx); in operator ()()
|
| D | conv_ops.cc | 1020 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()()
1051 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()()
1115 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()()
|
| D | ctc_loss_op.cc | 347 DnnScratchAllocator workspace_allocator(1LL << 32, ctx); in Compute()
|
| D | conv_ops_fused_impl.h | 395 DnnScratchAllocator scratch_allocator(ConvolveScratchSize(), context);
700 DnnScratchAllocator scratch_allocator(ConvolveScratchSize(), context);
|
| D | lrn_op.cc | 605 DnnScratchAllocator scratch_allocator(NormalizeBackwardScratchSize, in launch()
|