Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for ShapedType (0.14 sec)

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

    Type TypeMeet(Type lhs, Type rhs) {
      DCOMMENT("RefineTypeWith : " << lhs << " : " << rhs);
      if (lhs == rhs) return lhs;
    
      auto rhs_shape_type = mlir::dyn_cast<ShapedType>(rhs);
      if (!rhs_shape_type) return lhs;
      auto lhs_shape_type = mlir::cast<ShapedType>(lhs);
      if (lhs_shape_type.hasRank() && rhs_shape_type.hasRank() &&
          lhs_shape_type.getRank() != rhs_shape_type.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)
  2. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

        for (auto& dim : shape_ref) {
          // translate dynamic shapes from mlir to tfl values
          shape.push_back(
              dim == mlir::ShapedType::kDynamic ? 1 : static_cast<int>(dim));
          shape_signature.push_back(static_cast<int>(
              dim == mlir::ShapedType::kDynamic ? tensorflow::kTFDynamicSize
                                                : dim));
        }
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
Back to top