Searched refs:quant_op (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | depthwise_conv_test.cc | 725 QuantizedDepthwiseConvolutionOpModel quant_op( in TEST_P() local
747 quant_op.SetInput(input); in TEST_P()
748 quant_op.SetFilter(filter); in TEST_P()
749 quant_op.SetBias(bias); in TEST_P()
750 quant_op.Invoke(); in TEST_P()
757 EXPECT_THAT(quant_op.GetDequantizedOutput(), in TEST_P()
763 QuantizedDepthwiseConvolutionOpModel quant_op( in TEST_P() local
785 quant_op.SetInput(input); in TEST_P()
786 quant_op.SetFilter(filter); in TEST_P()
787 quant_op.SetBias(bias); in TEST_P()
[all …]
|
| D | conv_test.cc | 956 QuantizedConvolutionOpModel quant_op( in TEST_P() local
978 quant_op.SetInput(input); in TEST_P()
979 quant_op.SetFilter(filter); in TEST_P()
980 quant_op.SetBias(bias); in TEST_P()
981 quant_op.Invoke(); in TEST_P()
988 EXPECT_THAT(quant_op.GetDequantizedOutput(), in TEST_P()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | modify_model_interface_test.cc | 37 auto quant_op = absl::make_unique<OperatorT>(); in CreateQuantizedModelSingleInputOutput() local
62 quant_op->opcode_index = 0; in CreateQuantizedModelSingleInputOutput()
63 quant_op->inputs = {0}; in CreateQuantizedModelSingleInputOutput()
64 quant_op->outputs = {1}; in CreateQuantizedModelSingleInputOutput()
74 model->subgraphs[0]->operators.push_back(std::move(quant_op)); in CreateQuantizedModelSingleInputOutput()
425 auto quant_op = model->subgraphs[0]->operators[0].get(); in TEST() local
426 auto input = model->subgraphs[0]->tensors[quant_op->inputs[0]].get(); in TEST()
|
| D | quantize_model_test.cc | 227 const auto quant_op = quantized_graph->operators[i].get(); in TEST_P() local
229 EXPECT_EQ(quant_op->inputs, GetAsVector(float_op->inputs())); in TEST_P()
230 EXPECT_EQ(quant_op->outputs, GetAsVector(float_op->outputs())); in TEST_P()
231 EXPECT_EQ(quant_op->opcode_index, float_op->opcode_index()); in TEST_P()
271 const auto& quant_op = subgraph->operators[0]; in TEST_P() local
274 const int32_t quant_idx = quant_op->opcode_index; in TEST_P()
286 EXPECT_EQ(quant_op->inputs[0], input_idx); in TEST_P()
287 EXPECT_EQ(quant_op->outputs[0], readonly_subgraph->inputs()->Get(0)); in TEST_P()
299 EXPECT_EQ(subgraph->tensors[quant_op->outputs[0]]->name, in TEST_P()
1499 const auto& quant_op = subgraph->operators[0]; in TEST_P() local
[all …]
|
| D | quantize_model.cc | 749 std::unique_ptr<OperatorT> quant_op; in QuantizeOpInput() local
750 utils::MakeQuantizeOperator(model, &quant_op, tensor_idx, in QuantizeOpInput()
753 std::move(quant_op)); in QuantizeOpInput()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | util.py | 815 quant_op = schema_fb.OperatorCodeT()
816 quant_op.builtinCode = schema_fb.BuiltinOperator.QUANTIZE
817 quant_op.deprecatedBuiltinCode = schema_fb.BuiltinOperator.QUANTIZE
818 model.operatorCodes.append(quant_op)
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 769 ConvolutionOpModel quant_op({TensorType_UINT8, {2, 2, 4, 1}, -128.5, 128}, in TEST() local
790 quant_op.SetInput(input); in TEST()
791 quant_op.SetFilter(filter); in TEST()
792 quant_op.SetBias(bias); in TEST()
793 quant_op.Invoke(); in TEST()
800 EXPECT_THAT(quant_op.GetOutput(), in TEST()
1152 DepthwiseConvolutionOpModel quant_op( in TEST() local
1173 quant_op.SetInput(input); in TEST()
1174 quant_op.SetFilter(filter); in TEST()
1175 quant_op.SetBias(bias); in TEST()
[all …]
|