Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for GetCompatibleRankedTensorType (0.26 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

      return lhs.hasRank() && rhs.hasRank() && lhs.getRank() == rhs.getRank();
    }
    
    // Creates a compatible RankedTensorType where mismatched dimensions are
    // replaced with dynamic sizes.
    RankedTensorType GetCompatibleRankedTensorType(RankedTensorType lhs,
                                                   RankedTensorType rhs) {
      assert(lhs.getRank() == rhs.getRank());
      llvm::SmallVector<int64_t, 4> dims;
      dims.reserve(lhs.getRank());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
Back to top