- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 123 for input_dtype (0.26 sec)
-
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) -
subprojects/core/src/main/java/org/gradle/api/internal/initialization/transform/MergeInstrumentationAnalysisTransform.java
File input = getInput().get().getAsFile(); InstrumentationInputType inputType = getInputType(input); switch (inputType) { case DEPENDENCY_ANALYSIS_DATA: doMergeAndOutputAnalysis(input, outputs); return; case ORIGINAL_ARTIFACT:
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Thu Apr 18 15:08:33 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/ir/QuantOps.cc
for (auto input : llvm::zip(getOperandTypes(), getInputSpecs())) { Type inputType = std::get<0>(input); Attribute inputSpec = std::get<1>(input); if (!isValidQuantizationSpec(inputSpec, inputType)) { return emitOpError() << "has incompatible specification " << inputSpec << " and input type " << inputType; } } for (auto result : llvm::zip(getResultTypes(), getOutputSpecs())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/QuantOps.cc
for (auto input : llvm::zip(getOperandTypes(), getInputSpecs())) { Type inputType = std::get<0>(input); Attribute inputSpec = std::get<1>(input); if (!isValidQuantizationSpec(inputSpec, inputType)) { return emitOpError() << "has incompatible specification " << inputSpec << " and input type " << inputType; } } for (auto result : llvm::zip(getResultTypes(), getOutputSpecs())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K 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/tensorflow/ir/tf_ops_n_z.cc
<< ", and input of rank " << input_type.getRank(); } if (input_type && output_type) { if (input_type.getRank() != output_type.getRank()) { return op.emitOpError() << "expected rank of input to equal to rank of output" << ", got input of rank " << input_type.getRank() << ", and output of rank " << output_type.getRank(); }
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/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)