Home
last modified time | relevance | path

Searched defs:ConvParams (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_generic.h52 struct ConvParams { struct
53 DataType weights_data_type; // used for weights and biases
54 int4 block_size; // WHDS
55 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/Depth
59 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 …]
Dconv_metal_simd.h54 struct ConvParams { struct
55 int3 work_group_size;
56 int3 work_group_launch_order;
57 bool linear_spatial; // spatial dimensions are Width/Height/Depth
81 ConvParams params_; argument
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h697 struct ConvParams { struct
698 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 …]