- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getOutputShape (0.1 sec)
-
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
mlir::LogicalResult TransposeConvOp::verify() { TransposeConvOp op = *this; ShapedType output_type = op.getOutput().getType().cast<ShapedType>(); ShapedType output_shape_type = op.getOutputShape().getType().cast<ShapedType>(); if (output_type.hasRank() && output_shape_type.hasStaticShape()) { if (output_type.getRank() != output_shape_type.getDimSize(0)) { return op.emitOpError(llvm::formatv(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
PatternRewriter& rewriter) const override { rewriter.replaceOpWithNewOp<TFL::ReshapeOp>(op, op.getOperand(), op.getOutputShape()); } }; // Rewrites quantized stablehlo.select to tfl.select_v2. // TODO: b/322428814 - Add StableHLO quantizer integration tests for ODML.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0)