Searched refs:DnnScratchAllocator (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 39 class DnnScratchAllocator : public se::ScratchAllocator {
41 virtual ~DnnScratchAllocator() {} in ~DnnScratchAllocator()
42 DnnScratchAllocator(int64 memory_limit, OpKernelContext* context) in DnnScratchAllocator() function
|
| D | conv_ops_fused_impl.h | 526 DnnScratchAllocator scratch_allocator(ConvolveScratchSize(), context);
777 DnnScratchAllocator* scratch_allocator,
806 DnnScratchAllocator scratch_allocator(ConvolveScratchSize(), context);
|
| D | conv_ops_3d.cc | 454 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()
484 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()
|
| D | conv_grad_ops_3d.cc | 1370 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute()
1407 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute()
1776 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in Compute()
1814 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in Compute()
|
| D | conv_ops.cc | 864 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()()
894 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()()
|
| D | conv_grad_filter_ops.cc | 850 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in operator ()()
883 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, ctx); in operator ()()
|
| D | conv_grad_input_ops.cc | 929 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, ctx); in operator ()()
962 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in operator ()()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 718 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()
753 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()
|