- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for getDefiningOp (0.24 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize.cc
dyn_cast_or_null<TFL::ExpOp>(sum_op.getInput().getDefiningOp()); if (!exp_op || !exp_op->hasOneUse()) { return failure(); } auto parent_sub_op = dyn_cast_or_null<TFL::SubOp>(sub_op.getLhs().getDefiningOp()); if (!parent_sub_op || parent_sub_op != dyn_cast_or_null<TFL::SubOp>( exp_op.getX().getDefiningOp())) { return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
add_op != nullptr && !isa<stablehlo::ConstantOp>(add_op->getOperand(1).getDefiningOp())) { LLVM_DEBUG(llvm::dbgs() << "Expected a `stablehlo.constant` as the " << "rhs of `stablehlo.add`.\n"); } // Make sure the filter is a constant or a constant transpose. Operation* filter_op = filter.getDefiningOp(); const bool is_constant = isa_and_nonnull<stablehlo::ConstantOp>(filter_op);
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_a_m.cc
Operation* arg_op = arg.getDefiningOp(); return arg_op && arg_op->getName() == first_arg_op->getName(); }); if (!args_same_op) return failure(); // Collect unary operations operands. auto unary_operands = llvm::map_range(op.getValues(), [](Value arg) -> Value { return arg.getDefiningOp()->getOperand(0); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
while ( llvm::isa_and_nonnull<IdentityOp, IdentityNOp>(value.getDefiningOp())) { value = value.getDefiningOp()->getOperand( mlir::cast<OpResult>(value).getResultNumber()); } Operation* op = value.getDefiningOp(); if (!llvm::isa_and_nonnull<BatchDatasetV2Op, MapDatasetOp, RepeatDatasetOp,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
auto slice_op = dyn_cast_or_null<StridedSliceOp>(getValues()[0].getDefiningOp()); if (!slice_op) return {}; // Input to the slice op is defined by shape operation. auto shape_op = dyn_cast_or_null<ShapeOp>(slice_op.getInput().getDefiningOp()); if (!shape_op) return {}; // Input tensor, which shape is reconstructed by the pack operation.
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/ir/tfl_ops.cc
auto* defining_op = operand.getDefiningOp(); if (!llvm::dyn_cast_or_null<TF::BroadcastToOp>(defining_op) && !llvm::dyn_cast_or_null<TFL::BroadcastToOp>(defining_op)) { return nullptr; } Value broadcast_shape = defining_op->getOperand( 1); // Broadcasted shape operand of BroadcastTo op. Operation* parent_of_defining_op = broadcast_shape.getDefiningOp();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
// None type is allowed to represent unspecified operands. if (mlir::isa<NoneType>(value.getType())) return true; auto type = mlir::dyn_cast<TensorType>(value.getType()); if (!type) { if (auto op = value.getDefiningOp()) { error_handler.emitError() << '\'' << op << "' should produce value of tensor type instead of " << value.getType(); return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// Returns the first producer op whose type is not in Tys. template <typename... Tys> Value recursivelyWalkUp(Value op) const { while (llvm::isa_and_nonnull<Tys...>(op.getDefiningOp())) { Operation* producer = op.getDefiningOp(); op = producer->getOperand(/*idx=*/0); } return op; } }; class ConvertMaxPoolOp : public OpConversionPattern<mhlo::ReduceWindowOp> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
int dst_input) { // Get the NextIteration.Source operation from the token operand of the sink. mlir::Operation* source = sink->getOperand(0).getDefiningOp(); // Adds the "source" to the operands of the dst by creating a new dst // operation. mlir::OperationState state(dst->getLoc(), dst->getName()); auto num_operands = dst->getNumOperands();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)