Searched refs:input_datatype (Results 1 – 7 of 7) sorted by relevance
| /external/XNNPACK/tools/ |
| D | generate-vcvt-test.py | 34 input_datatype = match.group(1)
39 return input_datatype, output_datatype, batch_tile, arch, isa
246 def generate_test_cases(ukernel, init_fn, input_datatype, output_datatype, argument
271 "INPUT_DATATYPE": input_datatype.upper(),
308 input_datatype, output_datatype, batch_tile, arch, isa = \
315 name, init_fn, input_datatype, output_datatype, batch_tile, isa)
|
| /external/XNNPACK/src/subgraph/ |
| D | fully-connected.c | 192 enum xnn_datatype input_datatype, in validate_datatypes_with_bias() argument
199 if (input_datatype == xnn_datatype_fp32 && in validate_datatypes_with_bias()
208 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
218 if (input_datatype == xnn_datatype_quint8 && in validate_datatypes_with_bias()
233 enum xnn_datatype input_datatype, in validate_datatypes_without_bias() argument
239 if (input_datatype == xnn_datatype_fp32 && output_datatype == xnn_datatype_fp32) { in validate_datatypes_without_bias()
245 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
252 if (input_datatype == xnn_datatype_quint8 && output_datatype == xnn_datatype_quint8) { in validate_datatypes_without_bias()
|
| D | deconvolution-2d.c | 242 enum xnn_datatype input_datatype, in validate_datatypes_with_bias() argument
249 if (input_datatype == xnn_datatype_fp32 && in validate_datatypes_with_bias()
258 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
268 if (input_datatype == xnn_datatype_quint8 && in validate_datatypes_with_bias()
283 enum xnn_datatype input_datatype, in validate_datatypes_without_bias() argument
289 if (input_datatype == xnn_datatype_fp32 && output_datatype == xnn_datatype_fp32) { in validate_datatypes_without_bias()
295 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
302 if (input_datatype == xnn_datatype_quint8 && output_datatype == xnn_datatype_quint8) { in validate_datatypes_without_bias()
|
| D | depthwise-convolution-2d.c | 348 enum xnn_datatype input_datatype, in validate_datatypes_with_bias() argument
355 if (input_datatype == xnn_datatype_fp32 && in validate_datatypes_with_bias()
364 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
372 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
382 if (input_datatype == xnn_datatype_quint8 && in validate_datatypes_with_bias()
397 enum xnn_datatype input_datatype, in validate_datatypes_without_bias() argument
403 if (input_datatype == xnn_datatype_fp32 && output_datatype == xnn_datatype_fp32) { in validate_datatypes_without_bias()
409 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
414 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
421 if (input_datatype == xnn_datatype_quint8 && output_datatype == xnn_datatype_quint8) { in validate_datatypes_without_bias()
|
| D | convolution-2d.c | 347 enum xnn_datatype input_datatype, in validate_datatypes_with_bias() argument
354 if (input_datatype == xnn_datatype_fp32 && in validate_datatypes_with_bias()
363 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
371 if (input_datatype == xnn_datatype_qint8 && in validate_datatypes_with_bias()
381 if (input_datatype == xnn_datatype_quint8 && in validate_datatypes_with_bias()
396 enum xnn_datatype input_datatype, in validate_datatypes_without_bias() argument
402 if (input_datatype == xnn_datatype_fp32 && output_datatype == xnn_datatype_fp32) { in validate_datatypes_without_bias()
408 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
413 if (input_datatype == xnn_datatype_qint8 && output_datatype == xnn_datatype_qint8) { in validate_datatypes_without_bias()
420 if (input_datatype == xnn_datatype_quint8 && output_datatype == xnn_datatype_quint8) { in validate_datatypes_without_bias()
|
| D | convert.c | 166 enum xnn_datatype input_datatype, in validate_datatypes() argument
169 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()
|