Home
last modified time | relevance | path

Searched defs:pool_info (Results 1 – 25 of 60) sorted by relevance

123

/external/ComputeLibrary/src/runtime/CL/functions/
DCLROIPoolingLayer.cpp32 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
37 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
42 …ensor *input, const ICLTensor *rois, const ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
DCLROIAlignLayer.cpp34 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
41 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
46 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
DCLMaxUnpoolingLayer.cpp44 …nfigure(ICLTensor *input, ICLTensor *indices, ICLTensor *output, const PoolingLayerInfo &pool_info) in configure()
49 …ontext, ICLTensor *input, ICLTensor *indices, ICLTensor *output, const PoolingLayerInfo &pool_info) in configure()
58 …o *input, const ITensorInfo *indices, const ITensorInfo *output, const PoolingLayerInfo &pool_info) in validate()
DCLPooling3dLayer.cpp47 …g3dLayer::configure(const ICLTensor *input, ICLTensor *output, const Pooling3dLayerInfo &pool_info) in configure()
52 …t &compile_context, const ICLTensor *input, ICLTensor *output, const Pooling3dLayerInfo &pool_info) in configure()
61 …:validate(const ITensorInfo *input, const ITensorInfo *output, const Pooling3dLayerInfo &pool_info) in validate()
DCLPoolingLayer.cpp47 …oolingLayer::configure(ICLTensor *input, ICLTensor *output, const PoolingLayerInfo &pool_info, ICL… in configure()
52 …text &compile_context, ICLTensor *input, ICLTensor *output, const PoolingLayerInfo &pool_info, ICL… in configure()
62 …lidate(const ITensorInfo *input, const ITensorInfo *output, const PoolingLayerInfo &pool_info, con… in validate()
/external/ComputeLibrary/tests/validation/fixtures/
DROIAlignLayerFixture.h49 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
118 const ROIPoolingLayerInfo &pool_info, in compute_target()
166 const ROIPoolingLayerInfo &pool_info, in compute_reference()
193 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
205 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
DROIPoolingLayerFixture.h49 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
107 const ROIPoolingLayerInfo &pool_info, in compute_target()
151 const ROIPoolingLayerInfo &pool_info, in compute_reference()
178 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
191 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
/external/ComputeLibrary/src/core/CL/kernels/
DCLROIAlignLayerKernel.cpp45 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate_arguments()
85 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
90 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
147 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
DCLROIPoolingLayerKernel.cpp50 …o *input, const ITensorInfo *rois, const ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
74 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
79 …ensor *input, const ICLTensor *rois, const ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
DCLMaxUnpoolingLayerKernel.cpp45 …uments(const ITensorInfo *input, const ITensorInfo *output, const PoolingLayerInfo &pool_info, con… in validate_arguments()
80 …t ICLTensor *input, const ICLTensor *indices, ICLTensor *output, const PoolingLayerInfo &pool_info) in configure()
122 …o *input, const ITensorInfo *indices, const ITensorInfo *output, const PoolingLayerInfo &pool_info) in validate()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEROIAlignLayer.cpp32 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
39 …e(const ITensor *input, const ITensor *rois, ITensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
DNEROIPoolingLayer.cpp39 …o *input, const ITensorInfo *rois, const ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
44 …t ITensor *input, const ITensor *rois, const ITensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
DNEMaxUnpoolingLayer.cpp51 …er::configure(ITensor *input, ITensor *indices, ITensor *output, const PoolingLayerInfo &pool_info) in configure()
67 …o *input, const ITensorInfo *indices, const ITensorInfo *output, const PoolingLayerInfo &pool_info) in validate()
DNEPooling3dLayer.cpp52 …oling3dLayer::configure(const ITensor *input, ITensor *output, const Pooling3dLayerInfo &pool_info) in configure()
63 …:validate(const ITensorInfo *input, const ITensorInfo *output, const Pooling3dLayerInfo &pool_info) in validate()
DNEPoolingLayer.cpp53 void NEPoolingLayer::configure(ITensor *input, ITensor *output, const PoolingLayerInfo &pool_info, … in configure()
65 …lidate(const ITensorInfo *input, const ITensorInfo *output, const PoolingLayerInfo &pool_info, con… in validate()
/external/ComputeLibrary/src/cpu/operators/
DCpuMaxUnpooling.cpp32 … ITensorInfo *src, const ITensorInfo *indices, ITensorInfo *dst, const PoolingLayerInfo &pool_info) in configure()
40 …orInfo *src, const ITensorInfo *indices, const ITensorInfo *dst, const PoolingLayerInfo &pool_info) in validate()
DCpuPool3d.cpp45 …CpuPool3d::configure(const ITensorInfo *src, ITensorInfo *dst, const Pooling3dLayerInfo &pool_info) in configure()
55 …ol3d::validate(const ITensorInfo *src, const ITensorInfo *dst, const Pooling3dLayerInfo &pool_info) in validate()
DCpuPool2d.cpp51 void CpuPool2d::configure(ITensorInfo *src, ITensorInfo *dst, const PoolingLayerInfo &pool_info, IT… in configure()
91 …d::validate(const ITensorInfo *src, const ITensorInfo *dst, const PoolingLayerInfo &pool_info, con… in validate()
/external/ComputeLibrary/src/cpu/kernels/
DCpuMaxUnpoolingLayerKernel.cpp71 …orInfo *src, const ITensorInfo *indices, const ITensorInfo *dst, const PoolingLayerInfo &pool_info) in validate_arguments()
100 … ITensorInfo *src, const ITensorInfo *indices, ITensorInfo *dst, const PoolingLayerInfo &pool_info) in configure()
117 …orInfo *src, const ITensorInfo *indices, const ITensorInfo *dst, const PoolingLayerInfo &pool_info) in validate()
DCpuPool3dKernel.cpp68 …date_arguments(const ITensorInfo *src, const ITensorInfo *dst, const Pooling3dLayerInfo &pool_info) in validate_arguments()
121 …l3dKernel::configure(const ITensorInfo *src, ITensorInfo *dst, const Pooling3dLayerInfo &pool_info) in configure()
157 …rnel::validate(const ITensorInfo *src, const ITensorInfo *dst, const Pooling3dLayerInfo &pool_info) in validate()
DCpuPool2dKernel.cpp146 …lidate_arguments(const ITensorInfo *src, const ITensorInfo *dst, const PoolingLayerInfo &pool_info, in validate_arguments()
205 …window(ITensorInfo *src, ITensorInfo *dst, ITensorInfo *indices, const PoolingLayerInfo &pool_info, in validate_and_configure_window()
280 …Pool2dKernel::configure(ITensorInfo *src, ITensorInfo *dst, const PoolingLayerInfo &pool_info, ITe… in configure()
326 …l::validate(const ITensorInfo *src, const ITensorInfo *dst, const PoolingLayerInfo &pool_info, con… in validate()
/external/ComputeLibrary/src/core/NEON/kernels/
DNEROIAlignLayerKernel.cpp107 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate_arguments()
146 …e(const ITensor *input, const ITensor *rois, ITensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
170 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
DNEROIPoolingLayerKernel.cpp39 …o *input, const ITensorInfo *rois, const ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate_arguments()
115 …o *input, const ITensorInfo *rois, const ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
121 …t ITensor *input, const ITensor *rois, const ITensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
/external/ComputeLibrary/src/gpu/cl/kernels/
DClPool3dKernel.cpp45 …date_arguments(const ITensorInfo *src, const ITensorInfo *dst, const Pooling3dLayerInfo &pool_info) in validate_arguments()
96 …xt &compile_context, const ITensorInfo *src, ITensorInfo *dst, const Pooling3dLayerInfo &pool_info) in configure()
230 …rnel::validate(const ITensorInfo *src, const ITensorInfo *dst, const Pooling3dLayerInfo &pool_info) in validate()
/external/ComputeLibrary/tests/validation/reference/
DROIAlignLayer.cpp140 …impleTensor<float> &src, const SimpleTensor<float> &rois, const ROIPoolingLayerInfo &pool_info, co… in roi_align_layer()
213 … SimpleTensor<half> &src, const SimpleTensor<half> &rois, const ROIPoolingLayerInfo &pool_info, co… in roi_align_layer()
223 …Tensor<uint8_t> &src, const SimpleTensor<uint16_t> &rois, const ROIPoolingLayerInfo &pool_info, co… in roi_align_layer()
232 …eTensor<int8_t> &src, const SimpleTensor<uint16_t> &rois, const ROIPoolingLayerInfo &pool_info, co… in roi_align_layer()

123