Home
last modified time | relevance | path

Searched refs:output_datatype (Results 1 – 7 of 7) sorted by relevance

/external/XNNPACK/tools/
Dgenerate-vcvt-test.py36 output_datatype = match.group(3)
38 output_datatype = input_datatype
42 return input_datatype, output_datatype, batch_tile, arch, isa
270 def generate_test_cases(ukernel, init_fn, input_datatype, output_datatype, argument
296 "OUTPUT_DATATYPE": output_datatype.upper(),
332 input_datatype, output_datatype, batch_tile, arch, isa = \
339 name, init_fn, input_datatype, output_datatype, batch_tile, isa)
/external/XNNPACK/src/subgraph/
Dfully-connected.c224 enum xnn_datatype output_datatype) in validate_datatypes_with_bias() argument
230 output_datatype == xnn_datatype_fp32) in validate_datatypes_with_bias()
239 output_datatype == xnn_datatype_qint8) in validate_datatypes_with_bias()
249 output_datatype == xnn_datatype_quint8) in validate_datatypes_with_bias()
264 enum xnn_datatype output_datatype) in validate_datatypes_without_bias() argument
268 if (input_datatype == xnn_datatype_fp32 && output_datatype == xnn_datatype_fp32) { in validate_datatypes_without_bias()
274 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
281 if (input_datatype == xnn_datatype_quint8 && output_datatype == xnn_datatype_quint8) { in validate_datatypes_without_bias()
Dconvert.c205 enum xnn_datatype output_datatype) in validate_datatypes() argument
209 switch (output_datatype) { in validate_datatypes()
221 if (output_datatype == xnn_datatype_fp32) { in validate_datatypes()
226 switch (output_datatype) { in validate_datatypes()
236 switch (output_datatype) { in validate_datatypes()
Ddepthwise-convolution-2d.c356 enum xnn_datatype output_datatype) in validate_datatypes_with_bias() argument
362 output_datatype == xnn_datatype_fp32) in validate_datatypes_with_bias()
371 output_datatype == xnn_datatype_qint8) in validate_datatypes_with_bias()
379 output_datatype == xnn_datatype_qint8) in validate_datatypes_with_bias()
389 output_datatype == xnn_datatype_quint8) in validate_datatypes_with_bias()
404 enum xnn_datatype output_datatype) in validate_datatypes_without_bias() argument
408 if (input_datatype == xnn_datatype_fp32 && output_datatype == xnn_datatype_fp32) { in validate_datatypes_without_bias()
414 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
419 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
426 if (input_datatype == xnn_datatype_quint8 && output_datatype == xnn_datatype_quint8) { in validate_datatypes_without_bias()
Ddeconvolution-2d.c288 enum xnn_datatype output_datatype) in validate_datatypes_with_bias() argument
294 output_datatype == xnn_datatype_fp32) in validate_datatypes_with_bias()
303 output_datatype == xnn_datatype_qint8) in validate_datatypes_with_bias()
313 output_datatype == xnn_datatype_quint8) in validate_datatypes_with_bias()
328 enum xnn_datatype output_datatype) in validate_datatypes_without_bias() argument
332 if (input_datatype == xnn_datatype_fp32 && output_datatype == xnn_datatype_fp32) { in validate_datatypes_without_bias()
338 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
345 if (input_datatype == xnn_datatype_quint8 && output_datatype == xnn_datatype_quint8) { in validate_datatypes_without_bias()
Dconvolution-2d.c355 enum xnn_datatype output_datatype) in validate_datatypes_with_bias() argument
361 output_datatype == xnn_datatype_fp32) in validate_datatypes_with_bias()
370 output_datatype == xnn_datatype_qint8) in validate_datatypes_with_bias()
378 output_datatype == xnn_datatype_qint8) in validate_datatypes_with_bias()
388 output_datatype == xnn_datatype_quint8) in validate_datatypes_with_bias()
403 enum xnn_datatype output_datatype) in validate_datatypes_without_bias() argument
407 if (input_datatype == xnn_datatype_fp32 && output_datatype == xnn_datatype_fp32) { in validate_datatypes_without_bias()
413 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
418 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
425 if (input_datatype == xnn_datatype_quint8 && output_datatype == xnn_datatype_quint8) { in validate_datatypes_without_bias()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc4041 const xnn_datatype output_datatype = GetXNNPackDatatype(output_tensor); in VisitQuantizeNode() local
4049 if (input_datatype == output_datatype) { in VisitQuantizeNode()