Home
last modified time | relevance | path

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

123456

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A 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()
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 DClCol2ImKernel.cpp49 …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 DClIm2ColKernel.cpp63 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 DClWeightsReshapeKernel.cpp42 … 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 DCLChannelShuffleLayerKernel.cpp40 …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 DCLChannelShuffleLayer.cpp33 …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 DCLGEMMConvolutionLayer.cpp72 …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 DCLConvolutionLayer.cpp64 …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 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()
H A 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()
103 …const TensorShape &input_shape, const TensorShape &output_shape, unsigned int num_groups, DataType… in compute_reference()
H A 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()
118 …_shape, const TensorShape &output_shape, const bool has_bias, const unsigned int num_groups, DataT… in compute_reference()
H A DConvolutionLayerFixture.h68 … 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 Dgroup_assignment.h116 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 Des31cComputeShaderTests.cpp990 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 Dgl4cComputeShaderTests.cpp636 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 Dtests.rs119 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 Dqnormalization.cpp49 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 DNEChannelShuffleLayer.cpp32 …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 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
H A DFusedConvolutionBatchNormalizationWithPostOpsNode.cpp36unsigned int num_groups, in FusedConvolutionBatchNormalizationWithPostOpsNode()
75 unsigned int FusedConvolutionBatchNormalizationWithPostOpsNode::num_groups() const in num_groups() function in arm_compute::graph::FusedConvolutionBatchNormalizationWithPostOpsNode
H A DConvolutionLayerNode.cpp36 unsigned int num_groups, in ConvolutionLayerNode()
71 unsigned int ConvolutionLayerNode::num_groups() const in num_groups() function in arm_compute::graph::ConvolutionLayerNode
H A DFusedConvolutionBatchNormalizationNode.cpp36unsigned 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 Dnormalization.cpp11 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 Dttcmap.c1808 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 DCpuIm2ColKernel.cpp55 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()

123456