/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/ |
H A 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()
|
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/ |
H A 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()
|
H A 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()
|
H A 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()
|
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
H A 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()
|
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
H A 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()
|
H A 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()
|
H A 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()
|
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
H A 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()
|
H A 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()
|
H A 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()
|
H A D | ConvolutionLayerFixture.h | 68 … const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups) in configure_conv_function() 79 … const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups) in configure_conv_function() 213 const unsigned int num_groups = input_shape[2] / weights_shape[2]; variable 314 const unsigned int num_groups = input_shape[2] / weights_shape[2]; variable
|
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
H A 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
|
/aosp_15_r20/external/deqp/external/openglcts/modules/gles31/ |
H A D | es31cComputeShaderTests.cpp | 990 const GLuint num_groups[3] = {1, 1, 1}; in RunIteration() local 1098 std::string GenSource(const uvec3 &local_size, const uvec3 &num_groups) in GenSource() 1138 bool RunIteration(const uvec3 &local_size, const uvec3 &num_groups, bool dispatch_indirect) in RunIteration() 1286 std::string GenSource(const uvec3 &local_size, const uvec3 &num_groups) in GenSource() 1309 bool RunIteration(const uvec3 &local_size, const uvec3 &num_groups, bool dispatch_indirect) in RunIteration() 1462 std::string GenSource(const uvec3 &local_size, const uvec3 &num_groups) in GenSource() 1500 bool RunIteration(const uvec3 &local_size, const uvec3 &num_groups, bool dispatch_indirect) in RunIteration() 1686 std::string GenSource(const uvec3 &local_size, const uvec3 &num_groups) in GenSource() 1702 bool RunIteration(const uvec3 &local_size, const uvec3 &num_groups, bool dispatch_indirect) in RunIteration() 2029 std::string GenSource(const uvec3 &local_size, const uvec3 &num_groups) in GenSource() [all …]
|
/aosp_15_r20/external/deqp/external/openglcts/modules/gl/ |
H A D | gl4cComputeShaderTests.cpp | 636 const GLuint num_groups[3] = {1, 1, 1}; in RunIteration() local 745 std::string GenSource(const uvec3 &local_size, const uvec3 &num_groups) in GenSource() 785 bool RunIteration(const uvec3 &local_size, const uvec3 &num_groups, bool dispatch_indirect) in RunIteration() 940 std::string GenSource(const uvec3 &local_size, const uvec3 &num_groups) in GenSource() 979 bool RunIteration(const uvec3 &local_size, const uvec3 &num_groups, bool dispatch_indirect) in RunIteration() 1172 std::string GenSource(const uvec3 &local_size, const uvec3 &num_groups) in GenSource() 1190 bool RunIteration(const uvec3 &local_size, const uvec3 &num_groups, bool dispatch_indirect) in RunIteration() 1378 std::string GenSource(const uvec3 &local_size, const uvec3 &num_groups) in GenSource() 1394 bool RunIteration(const uvec3 &local_size, const uvec3 &num_groups, bool dispatch_indirect) in RunIteration() 2001 std::string GenSource(const uvec3 &local_size, const uvec3 &num_groups) in GenSource() [all …]
|
/aosp_15_r20/external/crosvm/ext2/tests/ |
H A D | tests.rs | 119 let num_groups = 2; in test_mkfs_empty_multi_block_groups() localVariable 639 let num_groups = 2; in test_multiple_bg_multi_inode_bitmap() localVariable 678 let num_groups = 4; in test_multiple_bg_multi_block_bitmap() localVariable 716 let num_groups = 50; in test_multiple_bg_big_files() localVariable
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
H A 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()
|
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
H A 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()
|
/aosp_15_r20/external/ComputeLibrary/src/graph/nodes/ |
H A 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
|
H A 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
|
H A 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
|
H A 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
|
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/ |
H A 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()
|
/aosp_15_r20/external/freetype/src/sfnt/ |
H A D | ttcmap.c | 1808 FT_UInt32 num_groups; in FT_CALLBACK_DEF() local 1909 FT_UInt32 num_groups = TT_NEXT_ULONG( p ); in tt_cmap8_char_index() local 1946 FT_UInt32 num_groups = TT_NEXT_ULONG( p ); in tt_cmap8_char_next() local 2255 FT_ULong num_groups; member 2285 FT_ULong num_groups; in tt_cmap12_validate() local 2409 FT_UInt32 num_groups = TT_PEEK_ULONG( p ); in tt_cmap12_char_map_binary() local 2599 FT_ULong num_groups; member 2629 FT_ULong num_groups; in tt_cmap13_validate() local 2730 FT_UInt32 num_groups = TT_PEEK_ULONG( p ); in tt_cmap13_char_map_binary() local
|
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
H A D | CpuIm2ColKernel.cpp | 55 bool has_bias, const Size2D &dilation, unsigned int num_groups) in validate_arguments() 335 bool has_bias, const Size2D &dilation, unsigned int num_groups) in configure() 426 bool has_bias, const Size2D &dilation, unsigned int num_groups) in validate()
|