Searched refs:input_datatype (Results 1 – 8 of 8) sorted by relevance
| /external/XNNPACK/tools/ |
| D | generate-vcvt-test.py | 34 input_datatype = match.group(1)
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
295 "INPUT_DATATYPE": input_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/ |
| D | fully-connected.c | 221 enum xnn_datatype input_datatype, in validate_datatypes_with_bias() argument
228 if (input_datatype == xnn_datatype_fp32 && in validate_datatypes_with_bias()
237 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
247 if (input_datatype == xnn_datatype_quint8 && in validate_datatypes_with_bias()
262 enum xnn_datatype input_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()
|
| D | depthwise-convolution-2d.c | 353 enum xnn_datatype input_datatype, in validate_datatypes_with_bias() argument
360 if (input_datatype == xnn_datatype_fp32 && in validate_datatypes_with_bias()
369 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
377 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
387 if (input_datatype == xnn_datatype_quint8 && in validate_datatypes_with_bias()
402 enum xnn_datatype input_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()
|
| D | deconvolution-2d.c | 285 enum xnn_datatype input_datatype, in validate_datatypes_with_bias() argument
292 if (input_datatype == xnn_datatype_fp32 && in validate_datatypes_with_bias()
301 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
311 if (input_datatype == xnn_datatype_quint8 && in validate_datatypes_with_bias()
326 enum xnn_datatype input_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()
|
| D | convolution-2d.c | 352 enum xnn_datatype input_datatype, in validate_datatypes_with_bias() argument
359 if (input_datatype == xnn_datatype_fp32 && in validate_datatypes_with_bias()
368 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
376 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
386 if (input_datatype == xnn_datatype_quint8 && in validate_datatypes_with_bias()
401 enum xnn_datatype input_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()
|
| D | convert.c | 204 enum xnn_datatype input_datatype, in validate_datatypes() argument
207 switch (input_datatype) { in validate_datatypes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_tfconversion_pass.cc | 387 DataType input_datatype; in RunPass() local
388 if ((GetNodeAttr(n->def(), "T", &input_datatype) == Status::OK()) && in RunPass()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 4040 const xnn_datatype input_datatype = GetXNNPackDatatype(input_tensor); in VisitQuantizeNode() local
4043 switch (input_datatype) { in VisitQuantizeNode()
4049 if (input_datatype == output_datatype) { in VisitQuantizeNode()
|