- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 83 for getDefiningOp (0.15 sec)
-
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc
auto bias_op = op->getOperand(bias_index).getDefiningOp<arith::ConstantOp>(); if (!affine_op || !bias_op || input_indices.size() != 2) return false; if (!mlir::isa<DenseFPElementsAttr>(bias_op.getValue())) return false; filter_index = affine_op.GetAffineOperandIndex(); if (!op->getOperand(filter_index).getDefiningOp<arith::ConstantOp>()) { return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 38.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc
return; } // Invariant: // isa<quantfork::QuantizeCastOp>(dq_arg.getDefiningOp()) --> // getdq_arg.getType() != q_op.getResult().getType() // // as otherwise qdq pair would have been optimized away. auto qd_arg_def_q_op = dyn_cast_or_null<quantfork::QuantizeCastOp>(dq_arg.getDefiningOp()); if (!qd_arg_def_q_op) { return; } qd_arg_def_q_op.emitWarning()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.cc
SmallVector<Value, 1> new_values; bool should_rewrite = false; for (auto operand : op.getArgs()) { if (auto corert_const_dense_tensor_op = operand.getDefiningOp<corert::ConstDenseTensorOp>()) { new_values.push_back( rewriter.create<fallback_async::ConstDenseTensorOp>( op.getLoc(), rewriter.getType<fallback::TFTensorType>(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 15.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dilated_conv.h
} // Check if the ConvOp's input is defined by `Expand` op, and the output used // by `Squeeze` op. Operation* producer_op = op.getOperand(0).getDefiningOp(); if (!producer_op || producer_op->getNumResults() != 1) { return rewriter.notifyMatchFailure( op, "op doesn't have a producer node that has a single result"); } if (!producer_op->hasOneUse() ||
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/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc
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/quantization/tensorflow/ops/tf_quantize_op.cc
ShapedType result_type, StringRef func_name, Value& func_input_arg) { Operation* input_op = input_val.getDefiningOp(); Operation* insertion_point = input_op->getParentOfType<func::FuncOp>(); if (!insertion_point) insertion_point = input_op->getParentOfType<ModuleOp>(); rewriter.setInsertionPointAfter(insertion_point);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h
// The input of the quantize op has already been quantized, i.e. // rescale. return failure(); } Operation* operand_op = operand.getDefiningOp(); if (operand_op == nullptr) { // When `QuantizeOpT`'s operand does not have a defining op, it means it // is a `BlockArgument`. The pattern does not match if there is no op to // quantize.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_to_vhlo.cc
auto new_value = converter.materializeArgumentConversion( rewriter, result.getLoc(), type, {result}); rewriter.replaceAllUsesExcept(result, new_value, new_value.getDefiningOp()); } // Wrap operands in an an unrealized cast to create a cast to buffer any type // changes to the operand, and apply type converter to operands: // V0 = op(operand) // ==>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 19:48:51 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc
return; } // Invariant: // isa<quantfork::QuantizeCastOp>(dq_arg.getDefiningOp()) --> // getdq_arg.getType() != q_op.getResult().getType() // // as otherwise qdq pair would have been optimized away. auto qd_arg_def_q_op = dyn_cast_or_null<quantfork::QuantizeCastOp>(dq_arg.getDefiningOp()); if (!qd_arg_def_q_op) { return; } qd_arg_def_q_op.emitWarning()
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/common/attrs_and_constraints.h
template <typename T = Operation*> Operation* FindOperandOfType(Operation* op) { for (Value operand_value : op->getOperands()) { if (isa<T>(operand_value.getDefiningOp())) { return operand_value.getDefiningOp(); } } return nullptr; } // Returns the function attribute for the given call op which is lifted for // quantization.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0)