| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLChannelShuffleLayer.cpp | 33 …LChannelShuffleLayer::configure(const ICLTensor *input, ICLTensor *output, unsigned int num_groups) in configure()
38 …ompileContext &compile_context, const ICLTensor *input, ICLTensor *output, unsigned int num_groups) in configure()
46 …huffleLayer::validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups) in validate()
|
| D | CLGEMMConvolutionLayer.cpp | 72 …const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups, const experi… in configure()
79 …fo, const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups, const ex… in configure()
118 …fo, const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups, const ex… in validate()
|
| D | CLConvolutionLayer.cpp | 64 …ion, const ActivationLayerInfo &act_info, bool enable_fast_math, unsigned int num_groups, const ex… in configure()
71 …ion, const ActivationLayerInfo &act_info, bool enable_fast_math, unsigned int num_groups, const ex… in configure()
130 …ion, const ActivationLayerInfo &act_info, bool enable_fast_math, unsigned int num_groups, const ex… in validate()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | ChannelShuffleLayerFixture.h | 49 … 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()
|
| D | Col2ImFixture.h | 52 …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()
103 …const TensorShape &input_shape, const TensorShape &output_shape, unsigned int num_groups, DataType… in compute_reference()
|
| D | WeightsReshapeFixture.h | 52 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()
118 …_shape, const TensorShape &output_shape, const bool has_bias, const unsigned int num_groups, DataT… in compute_reference()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEChannelShuffleLayerKernel.cpp | 41 …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()
161 …LayerKernel::validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups) in validate()
|
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClCol2ImKernel.cpp | 49 …st ITensorInfo *src, const ITensorInfo *dst, const Size2D &convolved_dims, unsigned int num_groups) in validate_arguments()
67 …e_window(ITensorInfo *src, ITensorInfo *dst, const Size2D &convolved_dims, unsigned int num_groups) in validate_and_configure_window()
93 …context, ITensorInfo *src, ITensorInfo *dst, const Size2D &convolved_dims, unsigned int num_groups) in configure()
134 …st ITensorInfo *src, const ITensorInfo *dst, const Size2D &convolved_dims, unsigned int num_groups) in validate()
|
| D | ClWeightsReshapeKernel.cpp | 42 … ITensorInfo *input, const ITensorInfo *biases, const ITensorInfo *output, unsigned int num_groups) in validate_arguments()
78 …text, const ITensorInfo *src, const ITensorInfo *biases, ITensorInfo *dst, unsigned int num_groups) in configure()
107 …const ITensorInfo *src, const ITensorInfo *biases, const ITensorInfo *dst, unsigned int num_groups) in validate()
|
| D | ClIm2ColKernel.cpp | 63 unsigned int num_groups) in validate_arguments()
96 …gned int num_elems_processed_per_iteration, bool is_padding_required_nchw, unsigned int num_groups) in validate_and_configure_window()
148 …ms, const PadStrideInfo &conv_info, bool has_bias, const Size2D &dilation, unsigned int num_groups) in configure_opencl_kernel()
306 unsigned int num_groups) in configure()
356 unsigned int num_groups) in validate()
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLChannelShuffleLayerKernel.cpp | 40 …us validate_arguments(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups) in validate_arguments()
103 …elShuffleLayerKernel::configure(const ICLTensor *input, ICLTensor *output, unsigned int num_groups) in configure()
108 …ompileContext &compile_context, const ICLTensor *input, ICLTensor *output, unsigned int num_groups) in configure()
176 …LayerKernel::validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups) in validate()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEChannelShuffleLayer.cpp | 32 …id NEChannelShuffleLayer::configure(const ITensor *input, ITensor *output, unsigned int num_groups) in configure()
40 …huffleLayer::validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups) in validate()
|
| D | NEGEMMConvolutionLayer.cpp | 58 …e2D &dilation, const ActivationLayerInfo &act_info, bool enable_fast_math, unsigned int num_groups) in configure()
78 …e2D &dilation, const ActivationLayerInfo &act_info, bool enable_fast_math, unsigned int num_groups) in validate()
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | ChannelShuffleLayerNode.cpp | 33 ChannelShuffleLayerNode::ChannelShuffleLayerNode(unsigned int num_groups) in ChannelShuffleLayerNode()
40 unsigned int ChannelShuffleLayerNode::num_groups() const in num_groups() function in arm_compute::graph::ChannelShuffleLayerNode
|
| D | FusedConvolutionBatchNormalizationWithPostOpsNode.cpp | 36 … unsigned int num_groups, in FusedConvolutionBatchNormalizationWithPostOpsNode()
75 unsigned int FusedConvolutionBatchNormalizationWithPostOpsNode::num_groups() const in num_groups() function in arm_compute::graph::FusedConvolutionBatchNormalizationWithPostOpsNode
|
| D | ConvolutionLayerNode.cpp | 36 unsigned int num_groups, in ConvolutionLayerNode()
71 unsigned int ConvolutionLayerNode::num_groups() const in num_groups() function in arm_compute::graph::ConvolutionLayerNode
|
| D | FusedConvolutionBatchNormalizationNode.cpp | 36 … unsigned int num_groups, in FusedConvolutionBatchNormalizationNode()
76 unsigned int FusedConvolutionBatchNormalizationNode::num_groups() const in num_groups() function in arm_compute::graph::FusedConvolutionBatchNormalizationNode
|
| D | FusedConvolutionWithPostOpNode.cpp | 36 unsigned int num_groups, in FusedConvolutionWithPostOpNode()
71 unsigned int FusedConvolutionWithPostOpNode::num_groups() const in num_groups() function in arm_compute::graph::FusedConvolutionWithPostOpNode
|
| /external/pytorch/torch/csrc/api/src/nn/options/ |
| D | normalization.cpp | 11 GroupNormOptions::GroupNormOptions(int64_t num_groups, int64_t num_channels) in GroupNormOptions()
20 GroupNormFuncOptions::GroupNormFuncOptions(int64_t num_groups) in GroupNormFuncOptions()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | group_assignment.h | 116 int num_groups() const { return global_.num_groups(); } in num_groups() function
125 StatusOr<int> num_groups(int slice_id) const { in num_groups() function
156 int num_groups() const { return replica_ids_.size(); } in num_groups() function
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qnormalization.cpp | 49 int num_groups = 1; // not relevant for LayerNorm in quantized_layer_norm_impl() local
58 int64_t num_groups, in quantized_group_norm_impl()
156 int64_t output_zero_point) { in TORCH_LIBRARY_IMPL()
|
| /external/tensorflow/tensorflow/python/util/ |
| D | lock_util_test.py | 29 def testGroups(self, num_groups): argument
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | Im2Col.cpp | 98 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/ |
| D | ConvolutionLayer.cpp | 46 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/ |
| D | GroupedConvolutionMutator.cpp | 46 …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
|