Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFTypeIsBFloat16OrHalfTensor (0.45 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/lite/transforms/dilated_conv.h 

      if (dilations && !TFIntListIsAllOnes(dilations)) {
    return rewriter.notifyMatchFailure(op, "dilations should be all 1");
  }

  if (!TFL::TFTypeIsFloat32Tensor(op.getInput()) &&
      !TFL::TFTypeIsBFloat16OrHalfTensor(op.getInput())) {
    return rewriter.notifyMatchFailure(
        op, "op's input is not float or half or bfloat16");
  }
  if (!TFL::TFDataFormatIsNHWC(op)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20K bytes
    - Viewed (0)
Back to top