Home
last modified time | relevance | path

Searched defs:num_groups (Results 1 – 25 of 94) sorted by relevance

1234

/external/ComputeLibrary/src/core/CL/kernels/
DCLCol2ImKernel.cpp45 …ensorInfo *input, const ITensorInfo *output, const Size2D &convolved_dims, unsigned int num_groups) in validate_arguments()
63 …dow(ITensorInfo *input, ITensorInfo *output, const Size2D &convolved_dims, unsigned int num_groups) in validate_and_configure_window()
92 …e(const ICLTensor *input, ICLTensor *output, const Size2D &convolved_dims, unsigned int num_groups) in configure()
97 …, const ICLTensor *input, ICLTensor *output, const Size2D &convolved_dims, unsigned int num_groups) in configure()
140 …ensorInfo *input, const ITensorInfo *output, const Size2D &convolved_dims, unsigned int num_groups) in validate()
DCLWeightsReshapeKernel.cpp38 … ITensorInfo *input, const ITensorInfo *biases, const ITensorInfo *output, unsigned int num_groups) in validate_arguments()
74 …figure(const ICLTensor *input, const ICLTensor *biases, ICLTensor *output, unsigned int num_groups) in configure()
79 …ntext, const ICLTensor *input, const ICLTensor *biases, ICLTensor *output, unsigned int num_groups) in configure()
117 … ITensorInfo *input, const ITensorInfo *biases, const ITensorInfo *output, unsigned int num_groups) in validate()
DCLChannelShuffleLayerKernel.cpp40 …us validate_arguments(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups) in validate_arguments()
93 …elShuffleLayerKernel::configure(const ICLTensor *input, ICLTensor *output, unsigned int num_groups) in configure()
98 …ompileContext &compile_context, const ICLTensor *input, ICLTensor *output, unsigned int num_groups) in configure()
151 …LayerKernel::validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups) in validate()
DCLIm2ColKernel.cpp59 unsigned int num_groups) in validate_arguments()
92 …gned int num_elems_processed_per_iteration, bool is_padding_required_nchw, unsigned int num_groups) in validate_and_configure_window()
158 …ms, const PadStrideInfo &conv_info, bool has_bias, const Size2D &dilation, unsigned int num_groups) in configure_opencl_kernel()
308 unsigned int num_groups) in configure()
315 unsigned int num_groups) in configure()
367 unsigned int num_groups) in validate()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLChannelShuffleLayer.cpp32 …LChannelShuffleLayer::configure(const ICLTensor *input, ICLTensor *output, unsigned int num_groups) in configure()
37 …ompileContext &compile_context, const ICLTensor *input, ICLTensor *output, unsigned int num_groups) in configure()
44 …huffleLayer::validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups) in validate()
DCLGEMMConvolutionLayer.cpp67 …gure(const ICLTensor *weights, const ICLTensor *biases, ICLTensor *output, unsigned int num_groups) in configure()
72 …ext, const ICLTensor *weights, const ICLTensor *biases, ICLTensor *output, unsigned int num_groups) in configure()
89 …TensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, unsigned int num_groups) in validate()
208 … const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups) in configure()
215 …weights_info, const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups) in configure()
433 …weights_info, const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups) in validate()
DCLConvolutionLayer.cpp50 …e2D &dilation, const ActivationLayerInfo &act_info, bool enable_fast_math, unsigned int num_groups) in configure()
57 …e2D &dilation, const ActivationLayerInfo &act_info, bool enable_fast_math, unsigned int num_groups) in configure()
103 …e2D &dilation, const ActivationLayerInfo &act_info, bool enable_fast_math, unsigned int num_groups) in validate()
/external/ComputeLibrary/tests/validation/fixtures/
DChannelShuffleLayerFixture.h49 … void setup(TensorShape shape, unsigned int num_groups, DataType data_type, DataLayout data_layout) in setup()
62 …TensorType compute_target(TensorShape shape, DataType data_type, unsigned int num_groups, DataLayo… in compute_target()
97 …eTensor<T> compute_reference(const TensorShape &shape, DataType data_type, unsigned int num_groups) in compute_reference()
DCol2ImFixture.h52 …nst unsigned int convolved_width, unsigned int convolved_height, unsigned int num_groups, DataType… in setup()
69 …, const TensorShape &output_shape, const Size2D &convolved_dims, unsigned int num_groups, DataType… in compute_target()
98 …const TensorShape &input_shape, const TensorShape &output_shape, unsigned int num_groups, DataType… in compute_reference()
DWeightsReshapeFixture.h52 void setup(TensorShape input_shape, DataType data_type, bool has_bias, unsigned int num_groups) in setup()
67 …_shape, const TensorShape &output_shape, const bool has_bias, const unsigned int num_groups, DataT… in compute_target()
108 …_shape, const TensorShape &output_shape, const bool has_bias, const unsigned int num_groups, DataT… in compute_reference()
DConvolutionLayerFixture.h55 … const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups) in configure_conv_function()
161 const unsigned int num_groups = input_shape[2] / weights_shape[2]; in compute_target() local
218 const unsigned int num_groups = input_shape[2] / weights_shape[2]; in compute_reference() local
/external/ComputeLibrary/src/core/NEON/kernels/
DNEChannelShuffleLayerKernel.cpp41 …us validate_arguments(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups) in validate_arguments()
64 void channel_shuffle_nhwc(const ITensor *input, ITensor *output, unsigned int num_groups, const Win… in channel_shuffle_nhwc()
90 void channel_shuffle_nchw(const ITensor *input, ITensor *output, unsigned int num_groups, const Win… in channel_shuffle_nchw()
141 …hannelShuffleLayerKernel::configure(const ITensor *input, ITensor *output, unsigned int num_groups) in configure()
165 …LayerKernel::validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups) in validate()
/external/mesa3d/src/mesa/state_tracker/
Dst_cb_compute.c39 const GLuint *num_groups, in st_dispatch_compute_common()
74 const GLuint *num_groups) in st_dispatch_compute()
89 const GLuint *num_groups, in st_dispatch_compute_group_size()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEChannelShuffleLayer.cpp32 …id NEChannelShuffleLayer::configure(const ITensor *input, ITensor *output, unsigned int num_groups) in configure()
39 …huffleLayer::validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups) in validate()
DNEIm2Col.cpp40 …ms, const PadStrideInfo &conv_info, bool has_bias, const Size2D &dilation, unsigned int num_groups) in configure()
49 unsigned int num_groups) in validate()
/external/ComputeLibrary/src/graph/nodes/
DChannelShuffleLayerNode.cpp33 ChannelShuffleLayerNode::ChannelShuffleLayerNode(unsigned int num_groups) in ChannelShuffleLayerNode()
40 unsigned int ChannelShuffleLayerNode::num_groups() const in num_groups() function in arm_compute::graph::ChannelShuffleLayerNode
DConvolutionLayerNode.cpp36 unsigned int num_groups, in ConvolutionLayerNode()
71 unsigned int ConvolutionLayerNode::num_groups() const in num_groups() function in arm_compute::graph::ConvolutionLayerNode
DFusedConvolutionBatchNormalizationNode.cpp36unsigned int num_groups, in FusedConvolutionBatchNormalizationNode()
76 unsigned int FusedConvolutionBatchNormalizationNode::num_groups() const in num_groups() function in arm_compute::graph::FusedConvolutionBatchNormalizationNode
/external/mesa3d/src/mesa/main/
Dcompute.c55 validate_DispatchCompute(struct gl_context *ctx, const GLuint *num_groups) in validate_DispatchCompute()
103 const GLuint *num_groups, in validate_DispatchComputeGroupSizeARB()
286 const GLuint num_groups[3] = { num_groups_x, num_groups_y, num_groups_z }; in dispatch_compute() local
353 const GLuint num_groups[3] = { num_groups_x, num_groups_y, num_groups_z }; in dispatch_compute_group_size() local
/external/mesa3d/src/mesa/drivers/dri/i965/
Dbrw_compute.c99 brw_dispatch_compute(struct gl_context *ctx, const GLuint *num_groups) { in brw_dispatch_compute()
132 const GLuint *num_groups, in brw_dispatch_compute_group_size()
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLGEMMConvolutionLayer.h116 void configure(const ICLTensor *input, const ICLTensor *biases, unsigned int num_groups) in configure()
127 …Context &compile_context, const ICLTensor *input, const ICLTensor *biases, unsigned int num_groups) in configure()
/external/tensorflow/tensorflow/python/util/
Dlock_util_test.py33 def testGroups(self, num_groups): argument
/external/ComputeLibrary/tests/validation/CL/
DIm2Col.cpp98 const unsigned int num_groups = 2; in TEST_CASE() local
110 const unsigned int num_groups = 2; in TEST_CASE() local
/external/ComputeLibrary/tests/validation/reference/
DConvolutionLayer.cpp46 const Size2D &dilation, unsigned int num_groups) in convolution_layer_nchw()
112 … const Size2D &dilation, unsigned int num_groups, QuantizationInfo out_quant_info) in convolution_layer()
/external/ComputeLibrary/src/graph/mutators/
DGroupedConvolutionMutator.cpp46 …Method method, ActivationLayerInfo fused_act, FastMathHint fast_math_hint, unsigned int num_groups) in create_grouped_convolution()
146 const unsigned int num_groups = conv_node->num_groups(); in mutate() local

1234