Searched defs:ConvParams (Results 1 – 6 of 6) sorted by relevance
52 struct ConvParams { struct53 DataType weights_data_type; // used for weights and biases54 int4 block_size; // WHDS55 bool fixed_work_group_size;56 int3 work_group_size;57 int3 work_group_launch_order;58 bool linear_spatial; // spatial dimensions are Width/Height/Depth59 bool linear_all; // linear_spatial & linear_all can not be used together,63 bool different_weights_for_height;64 bool groups_support = false; // convolution groups[all …]
54 struct ConvParams { struct55 int3 work_group_size;56 int3 work_group_launch_order;57 bool linear_spatial; // spatial dimensions are Width/Height/Depth81 ConvParams params_; argument
285 struct ConvParams { struct286 std::vector<T> stride;287 std::vector<T> padding;288 std::vector<T> dilation;289 bool transposed{};290 std::vector<T> output_padding;291 T groups{};292 bool benchmark{};293 bool deterministic{};294 bool cudnn_enabled{};[all …]
23 struct ConvParams { struct24 std::vector<int64_t> stride;25 std::vector<int64_t> padding;26 std::vector<int64_t> dilation;27 bool transposed;28 std::vector<int64_t> output_padding;29 int groups;30 bool benchmark;31 bool deterministic;
697 struct ConvParams { struct698 PaddingType padding_type;699 PaddingValues padding_values;701 int16_t stride_width;702 int16_t stride_height;703 int16_t dilation_width_factor;704 int16_t dilation_height_factor;707 int32_t input_offset;708 int32_t weights_offset;709 int32_t output_offset;[all …]
43 using ConvParams = ideep::convolution_forward_params; variable