Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFL_NumElementsTrait (0.24 sec)

  1. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

          CPred<"!$_op.getOperand(" # x #
            ").getType().cast<ShapedType>().hasStaticShape()">,
          CPred<"$_op.getOperand(" # x #
            ").getType().cast<ShapedType>().getNumElements() == " # n>]>;
    
    class TFL_NumElementsTrait<int x, int n> :
      PredOpTrait<"operand " # x # " has num of elements equals to  " # n,
        TFL_NumElements<x, n>>;
    
    // tf.uint8 and tf.quint8 are mapped to the same tflite types, so they are equal
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top