- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 253 for getWant (0.26 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td
(TFL_TopKV2Op $input, $k)>; def ReductionDimensionIsLastDim : Constraint<CPred<"($0.cast<IntegerAttr>().getInt() == " "$1.getType().cast<ShapedType>().getRank() - 1 || $0.cast<IntegerAttr>().getInt() == -1)">>; // Legalizes TF_ApproxTopKOp to TFL_TopKV2Op with the following constraints: // 1. It computes max k
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 28.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
if (!(tensor_type && tensor_type.hasRank())) { return op->emitError() << "cannot convert op with unranked or non-tensor input type " << tensor_type << "."; } int rank = tensor_type.getRank(); if (rank <= partition_dim) { return op->emitError() << "cannot partition " << first_operand_type << " (rank = " << rank << ") along dimension "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
if (!output_ty) return success(); int64_t expected_output_rank = std::max(x_ty.getRank(), y_ty.getRank()); if (output_ty.getRank() != expected_output_rank) return op.emitOpError() << "found invalid output rank, expected " << expected_output_rank << " but got " << output_ty.getRank(); // Check output batch dim with potential broadcasting.
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/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
auto input_shape = mlir::cast<ShapedType>(input.getType()); auto filter_shape = mlir::cast<ShapedType>(filter.getType()); if (!input_shape.hasRank() || input_shape.getRank() != 4 || !filter_shape.hasRank() || filter_shape.getRank() != 4) { emitError(loc, "input and filter are expected to be 4D tensors"); return {}; } const int feature_group_cnt =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
} const int64_t padding_nums_size = 2 * (rhs_shape.getRank() - 2); padding_nums.reserve(padding_nums_size); if (conv_padding.strref() == "EXPLICIT") { for (auto padding_elem : op.getExplicitPaddingAttr().template getAsRange<IntegerAttr>()) { padding_nums.push_back(padding_elem.getInt()); } } else if (conv_padding.strref() == "VALID") {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
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;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.td
def ZeroIntAttr : AttrConstraint<CPred<"$_self.cast<::mlir::IntegerAttr>().getInt() == 0">>; // Checks if the value has rank at most 'n'. class HasRankAtLeast<int n> : Constraint< CPred<"$0.getType().cast<ShapedType>().hasRank() && " "$0.getType().cast<ShapedType>().getRank() >= " # n>>; // Checks value is not produced by a TFL_Quant or // from TFL_Quant Op with same quant type.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc
ArrayAttr stride = mlir::dyn_cast<ArrayAttr>(identifier_to_attr["strides"]); const int stride_h = mlir::cast<IntegerAttr>(stride[1]).getInt(); const int stride_w = mlir::cast<IntegerAttr>(stride[2]).getInt(); return rewriter.getI64ArrayAttr({stride_h, stride_w}); } Attribute GetLhsDilationValue(PatternRewriter& rewriter,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
auto boxes_type = mlir::dyn_cast<RankedTensorType>(func_.getFunctionType().getInput(0)); if (boxes_type == nullptr || !boxes_type.hasRank() || boxes_type.getRank() != 2) { return func_.emitWarning() << "TFLite does not support batched input for " "non_max_suppression_padded"; } return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
platforms/core-execution/persistent-cache/src/main/java/org/gradle/cache/internal/SingleDepthFilesFinder.java
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Fri Sep 22 09:08:47 UTC 2023 - 2.9K bytes - Viewed (0)