Searched defs:custom_op_map (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
H A D | quantize_weights.cc | 96 const CustomOpMap& custom_op_map) { in GetWeightInputIndices() 140 const CustomOpMap& custom_op_map, int op_input_idx) { in IsQuantizedInput() 150 const CustomOpMap& custom_op_map, in IsHybridEvaluationOp() 188 const CustomOpMap& custom_op_map) { in CheckAllOpInputsQuantized() 212 const CustomOpMap& custom_op_map, in InsertQuantizableInputTensorsFromOperator() 360 const CustomOpMap& custom_op_map) { in PassQuantizationAndGetConsumers() 398 const CustomOpMap& custom_op_map, bool use_updated_hybrid_scheme, in QuantizeWeightsInt8() 612 CustomOpMap custom_op_map; in QuantizeWeights() local 627 CustomOpMap custom_op_map; in QuantizeWeights() local 646 CustomOpMap custom_op_map; in QuantizeWeights() local [all …]
|
H A D | quantize_weights_portable.cc | 82 const CustomOpMap& custom_op_map) { in GetWeightInputIndices() 126 const CustomOpMap& custom_op_map, int op_input_idx) { in IsQuantizedInput() 136 const CustomOpMap& custom_op_map, in IsHybridEvaluationOp() 174 const CustomOpMap& custom_op_map) { in CheckAllOpInputsQuantized() 198 const CustomOpMap& custom_op_map, in InsertQuantizableInputTensorsFromOperator() 346 const CustomOpMap& custom_op_map) { in PassQuantizationAndGetConsumers() 384 const CustomOpMap& custom_op_map, bool use_updated_hybrid_scheme, in QuantizeWeightsInt8() 598 CustomOpMap custom_op_map; in QuantizeWeights() local 613 CustomOpMap custom_op_map; in QuantizeWeights() local 631 CustomOpMap custom_op_map; in QuantizeWeights() local [all …]
|
H A D | quantize_weights_test.cc | 468 CustomOpMap custom_op_map; in TEST_F() local 516 CustomOpMap custom_op_map; in TEST_F() local 555 const CustomOpMap custom_op_map; in TEST_F() local 609 const CustomOpMap custom_op_map; in TEST_F() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
H A D | post_quantize.cc | 51 const quant::CustomOpMap& custom_op_map) in PostQuantizePass() 362 quant::CustomOpMap custom_op_map; member 412 bool emit_quant_adaptor_ops, const quant::CustomOpMap& custom_op_map) { in CreatePostQuantizePass()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/ |
H A D | quantize_weights_test.cc | 537 CustomOpMap custom_op_map; in TEST_F() local 584 CustomOpMap custom_op_map; in TEST_F() local 622 const CustomOpMap custom_op_map; in TEST_F() local 679 const CustomOpMap custom_op_map; in TEST_F() local
|
H A D | quantize_weights.cc | 81 const CustomOpMap& custom_op_map, int64_t minimum_elements_for_weights, in QuantizeWeights() 212 const CustomOpMap& custom_op_map, in QuantizeWeights()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
H A D | quantization_config.cc | 58 CustomOpMap& custom_op_map) { in ParseCustomOpSpecs()
|