- Sort Score
- Result 10 results
- Languages All
Results 21 - 27 of 27 for hasTask (0.11 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc
// dimensions. ShapedType shaped_type = mlir::dyn_cast<ShapedType>(call_op->getOperand(0).getType()); if (!shaped_type || !shaped_type.hasRank()) { return absl::InternalError("The input of BatchMatMul must have rank."); } } else if (function_name.contains("gather")) { // This op is guaranteed to be a constant as ODS checks IsConstTensor.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dilated_conv.h
return rewriter.notifyMatchFailure(op, "op's data format isn't NHWC"); } // Allow dynamic width and height dimensions only. auto result_ty = mlir::cast<TensorType>(op.getResult().getType()); if (!result_ty.hasRank() || result_ty.getRank() != 4 || result_ty.isDynamicDim(0) || result_ty.isDynamicDim(3)) { return rewriter.notifyMatchFailure( op, "only dynamic width and height dimensions are allowed"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h
for (const auto& axis_int : perm.getValues<APInt>()) { int64_t axis = axis_int.getSExtValue(); if (axis < 0) { axis += input_type.getRank(); } if (axis < 0 || (input_type.hasRank() && axis >= input_type.getRank())) { return transpose_op.emitOpError("perm must be in [-rank, rank)"); } if (std::count(axes.begin(), axes.end(), axis) > 0) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 28K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc
// Resolve constraints inferred from the tensor type. if (auto tensor = mlir::dyn_cast<TensorType>(value.getType())) { if (constraint == ValueConstraint::kRank && tensor.hasRank()) return success(); if (constraint == ValueConstraint::kShape && tensor.hasStaticShape()) return success(); } return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 27.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
ConversionPatternRewriter &rewriter) const override { Value lhs = adaptor.getLhs(); auto lhs_type = mlir::cast<ShapedType>(lhs.getType()); if (!lhs_type.hasRank()) { return rewriter.notifyMatchFailure( op, "Legalization supports cases where only lhs rank known."); } auto lhs_quant_type = GetUniformQuantizedType(
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/lite/experimental/tac/transforms/device_transform_patterns.cc
return failure(); auto input = splitv_op.getValue(); auto input_type = mlir::dyn_cast_or_null<RankedTensorType>(input.getType()); if (!input_type || !input_type.hasRank()) return failure(); for (auto result : splitv_op.getResults()) { auto result_type = mlir::dyn_cast<RankedTensorType>(result.getType()); if (result_type == nullptr) return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
auto resource_type = mlir::TF::ResourceType::get({resource_subtype}, builder.getContext()); auto tensor_type = mlir::cast<mlir::TensorType>(mlir_arg.getType()); if (tensor_type.hasRank()) { mlir_arg.setType( GetTypeFromTFTensorShape(tensor_type.getShape(), resource_type)); } else { mlir_arg.setType(mlir::UnrankedTensorType::get(resource_type)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0)