HomeSort by relevance Sort by last modified time
    Searched refs:fakequant_op (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
read_fake_quant_min_max.cc 51 auto* fakequant_op = static_cast<FakeQuantOperator*>(fakequant_base_op); local
55 if (!fakequant_op->minmax) {
56 CHECK_EQ(fakequant_op->inputs.size(), 3);
60 if (!IsConstantParameterArray(*model, fakequant_op->inputs[1])) {
66 const auto& min_array = model->GetArray(fakequant_op->inputs[1]);
67 const auto& max_array = model->GetArray(fakequant_op->inputs[2]);
70 fakequant_op->minmax.reset(new MinMax);
71 MinMax& minmax = *fakequant_op->minmax;
82 if (CountOpsWithInput(*model, fakequant_op->inputs[i]) == 1) {
83 model->EraseArray(fakequant_op->inputs[i])
    [all...]
drop_fake_quant.cc 34 auto* fakequant_op = static_cast<FakeQuantOperator*>(fakequant_base_op); local
36 if (!fakequant_op->minmax) {
40 const auto& output_array = model->GetArray(fakequant_op->outputs[0]);
46 for (int i = 1; i < fakequant_op->inputs.size(); i++) {
47 if (CountOpsWithInput(*model, fakequant_op->inputs[i]) == 1) {
48 model->EraseArray(fakequant_op->inputs[i]);
51 fakequant_op->inputs.resize(1);
resolve_constant_fake_quant.cc 34 const auto* fakequant_op = local
38 if (!fakequant_op->minmax) {
43 if (!IsConstantParameterArray(*model, fakequant_op->inputs[0])) {
47 const auto& input_array = model->GetArray(fakequant_op->inputs[0]);
48 auto& output_array = model->GetArray(fakequant_op->outputs[0]);
53 output_array.GetOrCreateMinMax() = *fakequant_op->minmax;
59 model->flags, *fakequant_op->minmax, &qparams);
69 if (CountOpsWithInput(*model, fakequant_op->inputs[0]) == 1) {
70 model->EraseArray(fakequant_op->inputs[0]);
dequantize.cc 152 auto* fakequant_op = new FakeQuantOperator; local
154 fakequant_op); local
160 fakequant_op->minmax.reset(new MinMax);
161 *fakequant_op->minmax = array->GetMinMax();
170 fakequant_op->inputs = {new_array_name};
171 fakequant_op->outputs = {array_name};
180 fakequant_op->inputs = {array_name};
181 fakequant_op->outputs = {new_array_name};
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 821 auto* fakequant_op = tensorflow_graph->add_node(); local
1874 FakeQuantOperator* fakequant_op = new FakeQuantOperator; local
    [all...]

Completed in 1951 milliseconds