- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 134 for input_type (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
llvm::SmallVector<int64_t, 4> input_shape(4, ShapedType::kDynamic); auto input_type = mlir::cast<TensorType>(op.getInput().getType()); if (input_type.hasRank()) { if (input_type.getRank() != 4) return op.emitOpError() << "requires input to be a 4D tensor, but got " << input_type; int64_t input_batch = input_type.getDimSize(0); if (input_batch != ShapedType::kDynamic &&
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/tensorflow/transforms/optimize.cc
if (!reshape_type.hasStaticShape()) return failure(); ArrayRef<int64_t> reshape_shape = reshape_type.getShape(); auto input_type = mlir::cast<ShapedType>(op.getInput().getType()); auto output_type = mlir::cast<ShapedType>(op.getOutput().getType()); if (!input_type.hasRank() || !output_type.hasRank()) return failure(); // The pattern attempts to reduce the rank of the input to BroadcastTo.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
return failure(); Value input = tf_op.getInput(); RankedTensorType input_type = mlir::dyn_cast<RankedTensorType>(input.getType()); // Only rank size four input will be only available by the tf.Conv2D // operator verification. if (!input_type || input_type.isDynamicDim(3)) { return failure(); } // Check if the given op is based on grouped convolution.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/ir/UniformSupport.h
// process. struct ExpressedToQuantizedConverter { // Creates a converter for the given input type. static ExpressedToQuantizedConverter forInputType(Type input_type); // Converts the inputType to be based on the given elemental type, // returning the new type (or nullptr and emit an error on failure). Type convert(quant::QuantizedType elemental_type) const; // Whether the conversion is legal.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h
if (dyn_cast_or_null<quantfork::QuantizeCastOp>(next_op)) return failure(); auto input_type = mlir::cast<ShapedType>(transpose_op.getInput().getType()); auto perm_type = mlir::cast<ShapedType>(transpose_op.getPerm().getType()); if (input_type.hasStaticShape() && perm_type.hasStaticShape()) { if (perm_type.getNumElements() != input_type.getRank()) { return transpose_op.emitOpError(
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/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
return failure(); } const auto input_type = op.getLhs().getType().cast<TensorType>(); if (!(input_type.getRank() == 2 || input_type.getRank() == 3)) { LLVM_DEBUG(llvm::dbgs() << "Input expected to have rank of 2 or 3. Got: " << input_type << ".\n"); return failure(); } const Value filter = op.getRhs();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc
BoolAttr narrow_range = builder.getBoolAttr(false); auto add_quantize_op = [&](Location loc, Type input_type, Block* block, Block::iterator insertion_point, Value arg, int i) { if (auto shaped = mlir::dyn_cast<ShapedType>(input_type)) { if (mlir::isa<FloatType>(shaped.getElementType())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
break; default: return nullptr; // Not yet supported } } else { return nullptr; // Not yet supported } input_type = input_type.clone(new_storage_type); return input_type; } // Replaces quant.qcast op to composite quantize_i8 function. class ReplaceQuantizePattern : public mlir::OpRewritePattern<quantfork::QuantizeCastOp> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.h
// applying the permutation to a given shape through a transpose. PermutationAndShape GetPermutationAndTransposedShape( llvm::ArrayRef<int64_t> permutation_array, ShapedType input_type, ConversionPatternRewriter& rewriter); // Create a single const integer. Value BuildIntConstOp(ImplicitLocOpBuilder& builder, ConversionPatternRewriter& rewriter, int64_t const_value,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 11:35:25 UTC 2023 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
auto output_type = RankedTensorType::get(output_shape, builder->getF32Type()); SmallVector<mlir::Type, 4> input_types{input_type, weight_type, bias_type, projection_type, layer_norm_scale_type}; auto func_type = builder->getFunctionType(input_types, output_type); auto func = func::FuncOp::create(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0)