- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 374 for gotType (0.22 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
StringAttr conv_padding = op.getPaddingAttr(); SmallVector<int64_t> padding_nums; ShapedType lhs_shape = mlir::cast<ShapedType>(op.getLhs().getType()); ShapedType rhs_shape = mlir::cast<ShapedType>(op.getRhs().getType()); // Handle only static shape cases. // TODO(b/260284866): Handle dynamic shape cases. if (!lhs_shape.hasStaticShape()) {
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/transforms/decompose_hybrid_quantization.cc
SmallVector<Value> newOperands; newOperands.reserve(op->getNumOperands()); for (auto operand : op->getOperands()) { if (QuantizedType::getQuantizedElementType(operand.getType())) { auto newTy = QuantizedType::castToExpressedType(operand.getType()); newOperands.push_back( rewriter.create<TFL::DequantizeOp>(loc, newTy, operand)); continue; } newOperands.push_back(operand); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/arithmetic_count_util.h
auto weight = op->getOperand(1); auto weight_type = mlir::dyn_cast_or_null<mlir::RankedTensorType>(weight.getType()); if (weight_type == nullptr || !weight_type.hasStaticShape()) return false; auto output = op->getResult(0); auto output_type = mlir::dyn_cast_or_null<mlir::RankedTensorType>(output.getType()); if (output_type == nullptr || !output_type.hasStaticShape()) return false; int64_t cols = 1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc
<< "operand #" << i << " does not have a graph results to bind"; if (graph.getResult(i).getType() != operand.getType()) { return fetch.emitOpError() << "operand #" << i << " type mismatch graph results (" << graph.getResult(i).getType() << " != " << operand.getType() << ")"; } } return success(); } void GraphOp::print(OpAsmPrinter &p) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc
auto current_type = quant::QuantizedType::getQuantizedElementType( quantize_output.getType()) .getStorageType(); if (current_type == input_type) { // int8 == int8 arg_type = quantize_output.getType(); new_arg = block.addArgument(arg_type, loc); quantize_output.replaceAllUsesWith(new_arg);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.cc
// etc. constexpr int64_t kLookupTableFindCostScale = 8; constexpr int64_t kLookupTableFindStringKeyCostScale = 16; auto value_type = mlir::cast<mlir::TensorType>(op.getValues().getType()); auto key_type = mlir::cast<mlir::TensorType>(op.getKeys().getType()); int64_t output_size = InferTensorSize(context, value_type); int64_t cost = kLookupTableFindCostScale * output_size;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
"TFL::IsBroadcastableElementsAttrAndType($0.getType(), $1.getType())">>; def OperandsBroadcastToOutputType : Constraint<CPred< "TFL::OperandsBroadcastToOutputType($0.getType(), $1.getType(), " "$2.getType())">>; def IsTailOfShape : Constraint<CPred< "TFL::IsTailOfShape($0.getType(), $1.getType())">>; def IsReducedTailOfShape : Constraint<CPred<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
return false; } auto opr1_type = llvm::dyn_cast_or_null<RankedTensorType>(op->getOperand(0).getType()); auto opr2_type = llvm::dyn_cast_or_null<RankedTensorType>(op->getOperand(1).getType()); auto res_type = llvm::dyn_cast_or_null<RankedTensorType>(op->getResult(0).getType()); if (!(opr1_type && opr2_type && res_type)) { return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc
if (!input_dequant) return failure(); if (!IsQI32Type(input_dequant.getType())) return failure(); auto output_type = mlir::dyn_cast_or_null<ShapedType>(dequant_op.getOutput().getType()); if (!output_type || !output_type.getElementType().isF32()) return failure(); auto input_type = mlir::dyn_cast<ShapedType>(input_dequant.getType()); // TODO(renjieliu): support UniformQuantizedPerAxisType.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc
auto dcast = b.create<quantfork::DequantizeCastOp>( dq.getLoc(), dq.getOutput().getType(), dq.getInput()); dq.getOutput().replaceAllUsesWith(dcast); dq.erase(); } else if (auto q = llvm::dyn_cast<QuantizeOp>(op)) { auto qcast = b.create<quantfork::QuantizeCastOp>( q.getLoc(), q.getOutput().getType(), q.getInput()); q.getOutput().replaceAllUsesWith(qcast); q.erase();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 02:50:01 UTC 2024 - 3.5K bytes - Viewed (0)