Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for isIntOrIndexOrFloat (2.97 sec)

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

      // This is the usual case, but consider falling back to old approach
      // if transposing string tensors becomes needed while folding.
      if (!input_tensor.getElementType().isIntOrIndexOrFloat()) return nullptr;
      SmallVector<int64_t> perms;
      SmallVector<int64_t> output_shape;
      ArrayRef<int64_t> input_shape = input_tensor.getType().getShape();
      auto attr_iter = perm_tensor.getValues<IntegerAttr>();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
Back to top