- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 78 for getDefiningOp (0.24 sec)
-
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.td
class AreIntSplatValuesEqual<string type> : Constraint<CPred< "AreSplatValuesEqual<"# type #">($0, $1)">>; // Returns defining op of this value. def GetDefiningOp : NativeCodeCall<"$0.getDefiningOp()">; // Clones an operation with new operands while keeping attributes. def CloneOpWithReplacedOperands : NativeCodeCall< "CloneOpWithReplacedOperands("
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 04:55:44 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/rewrite_quantized_io.cc
if (!returned_type || !returned_type.getElementType().isa<quant::QuantizedType>()) { continue; } if (auto returned_op = returned_value.get().getDefiningOp<TFR::CastOp>()) { auto new_type = returned_type.clone(returned_type.getElementType() .cast<quant::QuantizedType>()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_splits.cc
for (Value operand : op->getOperands()) { if (Operation* pred = operand.getDefiningOp()) { int result_number = llvm::cast<OpResult>(operand).getResultNumber(); if (auto pred_island = llvm::dyn_cast<tf_executor::IslandOp>(pred)) { Value yield_operand = pred_island.GetYield().getOperand(result_number); predecessors.push_back(yield_operand.getDefiningOp()); } } } return predecessors; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 18:44:34 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
LogicalResult matchAndRewrite(TF::LookupTableFindV2Op find_op, PatternRewriter& rewriter) const override { auto handle_op = find_op.getTableHandle().getDefiningOp(); if (handle_op == nullptr) return failure(); auto hashtable_op = llvm::dyn_cast<TFL::HashtableOp>(handle_op); if (hashtable_op == nullptr) return failure(); rewriter.replaceOpWithNewOp<TFL::HashtableFindOp>(
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/tfrt/runtime_fallback/runtime_fallback_combine.cc
mlir::tfd::MoveDhtToTftOp move_input_op_0 = llvm::dyn_cast_or_null<mlir::tfd::MoveDhtToTftOp>( convert_op_input_0.getDefiningOp()); mlir::tfd::MoveDhtToTftOp move_input_op_1 = llvm::dyn_cast_or_null<mlir::tfd::MoveDhtToTftOp>( convert_op_input_1.getDefiningOp()); // The inputs should be MoveDhtToTftOp. if (!move_input_op_0 || !move_input_op_1) return mlir::failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 09 12:09:19 UTC 2022 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_variables.td
include "mlir/IR/PatternBase.td" include "tensorflow/compiler/mlir/lite/ir/tfl_ops.td" include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td" def HasSupportedElementType : Constraint<CPred<"HasSupportedElementType($0.getDefiningOp())">>; def IsSupportedElementType : Constraint<CPred<"IsSupportedElementType($0.getType().cast<ShapedType>())">>; def LegalizeVarHandle : Pat< (TF_VarHandleOp:$result $container, $shared_name),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 16 23:20:46 UTC 2022 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_device_copy_conversion.cc
// * the arg to the identity op is produced by a TPUExecuteOp. if (op_device && op_device.getValue().contains("TPU")) { return true; } Operation *def_op = arg.getDefiningOp(); // If the arg to this identity op is the arg of a function, there's no // defining op. if (def_op != nullptr && (isa<TF::TPUExecuteOp, TF::TPUExecuteAndUpdateVariablesOp>(def_op))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_dataflow.cc
ResourceConstructingOps result(global_tensor); return result; } } else if (auto vh = dyn_cast<TF::VarHandleOp>(value.getDefiningOp())) { return ResourceConstructingOps(vh); } else if (auto it = dyn_cast<TF::IteratorOp>(value.getDefiningOp())) { return ResourceConstructingOps(it); } return ResourceConstructingOps(); } ResourceConstructingOps ResourceConstructingOps::join(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
} if (v.getDefiningOp()->getNumOperands() == 1) { // If the value is originating from an unary op, assume it's something // simple like "cast" and keep tracing. return traceUpwardsToArgument(v.getDefiningOp()->getOperand(0), seen, out); } else { // Typically a tf.VarHandle op. return v.getDefiningOp()->emitOpError("Non constant predecessor"); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/remove_var_init_by_const.cc
Value resource_operand = assign_op.getOperand(0); Value assigned_value_operand = assign_op.getOperand(1); if (isa<TF::VarHandleOp>(resource_operand.getDefiningOp()) && isa<TF::ConstOp>(assigned_value_operand.getDefiningOp())) { return success(); } else { return failure(); } } void rewrite(TF::AssignVariableOp assign_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 03 12:04:03 UTC 2023 - 4.5K bytes - Viewed (0)