- Sort Score
- Result 10 results
- Languages All
Results 81 - 83 of 83 for getDefiningOp (0.12 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
if (!buffer_ty.hasStaticShape()) { mlir::Value return_val = return_op->getOperand(idx); if (auto owner = mlir::dyn_cast_or_null<mlir::tensor::CastOp>( return_val.getDefiningOp())) { // For bounded dynamic type, get a static size by taking bounds as the // dimensions. These dimensions are marked as dynamic in xla::Shape // below.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
// If `value` is produced by tf.Dequantize op, then return the Dequantize op's // input. Otherwise return `value`. auto get_real_input_value = [](Value value) -> Value { Operation* defining_op = value.getDefiningOp(); if (auto dequantize = dyn_cast_or_null<TF::DequantizeOp>(defining_op)) { return dequantize.getInput(); } else if (auto dequantize = dyn_cast_or_null<TFL::DequantizeOp>(defining_op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
// Extend tuple if `value` is a tuple. // If `value` is an op result and the owner is a `mhlo.tuple`, simply unpack // the tuple. if (auto tuple_op = value.getDefiningOp<TupleOp>()) { auto tuple_operands = llvm::to_vector(tuple_op.getOperands()); tuple_operands.push_back(token); return {create_tuple(tuple_operands)}; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0)