Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for HasRankAtMost (0.38 sec)

  1. tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td

       (HasRankAtMost<4> $input),
       (HasRankAtMost<4> $operand)]>;
    
      def FuseTileBroadcastToBinaryOp2#BinaryOp : Pat<
        (BinaryOp:$result $operand,
          (TFL_TileOp $input, (Arith_ConstantOp $tile)), $act_func),
        (BinaryOp $operand, $input, $act_func),
      [(OperandsBroadcastToOutputType $operand, $input, $result),
       (HasRankAtMost<4> $operand),
       (HasRankAtMost<4> $input)]>;
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 20:31:41 UTC 2024
    - 66.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_tensor_helper.h

      RankedTensorType type = GetRankedTensorTypeForOperand(value);
      return !type || type.getRank() >= rank;
    }
    
    // Returns true if the given `value` has at most the specified rank or has
    // unranked type.
    inline bool HasRankAtMost(Value value, int64_t rank) {
      RankedTensorType type = GetRankedTensorTypeForOperand(value);
      return !type || type.getRank() <= rank;
    }
    
    inline bool IsUnknownDimOrRank(int64_t dim_or_rank) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      if (!ttype.hasRank()) return success();
      if (!HasRankAtLeast(op.getInput(), 2))
        return op.emitOpError(
            "requires ranked input tensor to be of rank 2 or more");
      if (!HasRankAtMost(op.getInput(), std::numeric_limits<int32_t>::max()))
        return op.emitOpError(
            "requires ranked input tensor to be of rank INT32_MAX or less");
    
      return success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      Operation::operand_range values = op.getValues();
    
      int axis_idx = std::is_same<OpT, ConcatOp>() ? 0 : 1;
      Value axis = *op.getODSOperands(axis_idx).begin();
      if (!HasRankAtMost(axis, 1)) {
        return op.emitOpError(
            "requires axis to be of scalar type (or vector type for older "
            "versions)");
      }
    
      return VerifyTypesCompatibility(values,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/optimize.cc

    //       $qparams)),
    //   $act_fn),
    //   [(AnyStaticShapeTensor $lhs),
    //    (IsLastDimEqualToNumElements $bias, $bias),
    //    (HasOneUse $const_op),
    //    (HasRankAtMost<4> $bias),
    //    (HasRankAtLeast<2> $bias),
    //    (IsDefinedByFullyConnectedOp $lhs)]>;
    struct UndoBroadcastFullyConnectedBiasAddWithQDQs
        : public OpRewritePattern<TFL::AddOp> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top