- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for HasRankAtMost (0.38 sec)
-
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) -
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) -
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) -
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) -
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)