Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for quantize_axis (0.21 sec)

  1. tensorflow/compiler/mlir/lite/quantization/import_quant_stats_pass.cc

        axis_stats = DenseFPElementsAttr::get(
            RankedTensorType::get(axis_stats_shape, b.getF32Type()), min_maxs);
        axis = b.getI64IntegerAttr(info.meta().quantize_axis());
      }
    
      b.setInsertionPointAfter(op);
      if (IsQuantizableResult(op, index)) {
        InsertStatsOpAtResult(b, op->getResult(index), layer_stats, axis_stats,
                              axis);
      } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 08 10:41:08 UTC 2024
    - 9.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

                                tensor_shape_pb2.TensorShapeProto.Dim(
                                    size=filter_shape[quantized_axis]
                                )
                            ]
                        )
                    ]
                )
            )
          else:
            quantized_axis = -1
            # Empty dimension. Per-tensor quantization has singular channel.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
Back to top