Searched refs:quantize_op (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | post_quantize.cc | 73 auto remove_quantize_op = [&](QuantizeOp quantize_op) { in RemoveQuantizationAdaptorOps() argument
74 auto quantize_output = quantize_op.output(); in RemoveQuantizationAdaptorOps()
79 quantize_op.erase(); in RemoveQuantizationAdaptorOps()
86 auto quantize_op = llvm::cast<QuantizeOp>(*arg.user_begin()); in RemoveQuantizationAdaptorOps() local
87 remove_quantize_op(quantize_op); in RemoveQuantizationAdaptorOps()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 534 QuantizeV2 quantize_op(root.WithOpName("quantize_op"), dequantize_op, in TestRemoveRedundantQuantization() local
539 Dequantize(root.WithOpName("final_dequantize"), quantize_op.output, in TestRemoveRedundantQuantization()
540 quantize_op.output_min, quantize_op.output_max); in TestRemoveRedundantQuantization()
633 QuantizeV2 quantize_op(root.WithOpName("quantize_op"), dequantize_op, in TestRemoveRedundantQuantizationWithBiasAdd() local
638 Dequantize(root.WithOpName("final_dequantize"), quantize_op.output, in TestRemoveRedundantQuantizationWithBiasAdd()
639 quantize_op.output_min, quantize_op.output_max); in TestRemoveRedundantQuantizationWithBiasAdd()
759 QuantizeV2 quantize_op(root.WithOpName("quantize_op"), dequantize_op, in TestRemoveRedundantQuantizationWithMultipleOutputs() local
764 Dequantize(root.WithOpName("final_dequantize"), quantize_op.output, in TestRemoveRedundantQuantizationWithMultipleOutputs()
765 quantize_op.output_min, quantize_op.output_max); in TestRemoveRedundantQuantizationWithMultipleOutputs()
984 QuantizeV2 quantize_op(root.WithOpName("quantize_op"), dequantize_op, in TestMergeAdjacentRequantizes() local
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | modify_model_interface_test.cc | 630 auto quantize_op = model->subgraphs[0]->operators[2].get(); in TEST() local
631 auto output = model->subgraphs[0]->tensors[quantize_op->outputs[0]].get(); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 6316 "quantize_op.cc",
6494 "quantize_op.cc",
7520 "quantize_op.cc",
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 560 for (const string& quantize_op : *GetQuantizeOps(&validator)) { in TEST_F()
561 QCHECK(op_validators(&validator).count(quantize_op)); in TEST_F()
|