Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_metal.h40 struct ConvParams { struct
41 int3 block_size;
42 int3 work_group_size;
43 int3 work_group_launch_order;
44 int src_depth_loop_size;
45 bool need_src_loop = true;
46 bool need_dst_loop = true;
47 bool linear_wh;
48 bool linear_whs;
49 WeightsUploadType weights_upload_type;
[all …]
Dconv_powervr.h73 struct ConvParams { struct
81 DataType weights_data_type; // used for weights and biases
82 int4 block_size; // WHDS
83 bool fixed_work_group_size;
84 bool linear_spatial; // spatial dimensions are Width/Height/Depth
85 bool linear_all; // linear_spatial & linear_all can not be used together,
89 bool different_weights_for_height;
90 int src_depth_loop_size;
91 WeightsUploadType weights_upload_type;
92 bool x_kernel_is_1;
[all …]
Dconv_buffer_1x1.h58 struct ConvParams { struct
69 ConvBuffer1x1(const OperationDef& definition, const ConvParams& conv_params); argument
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h852 struct ConvParams { struct
853 PaddingType padding_type;
854 PaddingValues padding_values;
856 int16_t stride_width;
857 int16_t stride_height;
858 int16_t dilation_width_factor;
859 int16_t dilation_height_factor;
862 int32_t input_offset;
863 int32_t weights_offset;
864 int32_t output_offset;
[all …]