Home
last modified time | relevance | path

Searched refs:dequantize_op (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dremove_final_dequantize_op.cc33 const auto* dequantize_op = dequantize_it->get(); in Run() local
34 if (dequantize_op->type != OperatorType::kDequantize) { in Run()
37 const auto& output = dequantize_op->outputs[0]; in Run()
51 model->flags.set_output_arrays(i, dequantize_op->inputs[0]); in Run()
56 AddMessageF("Removed final %s", LogName(*dequantize_op)); in Run()
Dquantize.cc539 auto* dequantize_op = dequantize_it->get(); in Run() local
540 CHECK(dequantize_op->type == OperatorType::kDequantize); in Run()
541 op.inputs[input_index] = dequantize_op->inputs[0]; in Run()
544 if (!CountOpsWithInput(*model, dequantize_op->outputs[0])) { in Run()
545 if (IsDiscardableArray(*model, dequantize_op->outputs[0])) { in Run()
547 model->EraseArray(dequantize_op->outputs[0]); in Run()
555 dequantize_op->outputs[0]) { in Run()
557 if (IsInputArray(*model, dequantize_op->inputs[0])) { in Run()
566 i, LogName(*dequantize_op), in Run()
568 dequantize_op->inputs[0]); in Run()
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc507 Output dequantize_op = in TestRemoveRedundantQuantization() local
524 dequantize_op, dequantize_reshape_dims); in TestRemoveRedundantQuantization()
534 QuantizeV2 quantize_op(root.WithOpName("quantize_op"), dequantize_op, in TestRemoveRedundantQuantization()
606 Output dequantize_op = in TestRemoveRedundantQuantizationWithBiasAdd() local
623 dequantize_op, dequantize_reshape_dims); in TestRemoveRedundantQuantizationWithBiasAdd()
633 QuantizeV2 quantize_op(root.WithOpName("quantize_op"), dequantize_op, in TestRemoveRedundantQuantizationWithBiasAdd()
732 Output dequantize_op = in TestRemoveRedundantQuantizationWithMultipleOutputs() local
749 dequantize_op, dequantize_reshape_dims); in TestRemoveRedundantQuantizationWithMultipleOutputs()
759 QuantizeV2 quantize_op(root.WithOpName("quantize_op"), dequantize_op, in TestRemoveRedundantQuantizationWithMultipleOutputs()
767 Output relu_op = Relu(root.WithOpName("relu_op"), dequantize_op); in TestRemoveRedundantQuantizationWithMultipleOutputs()
[all …]
Dfreeze_requantization_ranges_test.cc96 Output dequantize_op = in TestFreezeRequantizationRanges() local
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights.cc323 std::unique_ptr<OperatorT> dequantize_op; in QuantizeWeightsInternal() local
324 MakeDequantizeOperator(model.get(), &dequantize_op, tensor_idx, in QuantizeWeightsInternal()
341 std::move(dequantize_op)); in QuantizeWeightsInternal()
/external/tensorflow/tensorflow/contrib/makefile/
Dtf_op_files.txt115 tensorflow/core/kernels/dequantize_op.cc
/external/tensorflow/tensorflow/core/kernels/
DBUILD5849 "dequantize_op.cc",
6030 "dequantize_op.cc",