- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 29 for new_operands (0.2 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
// AddLoadsStoresOutsideControlFlowOp(). auto new_operands = FilterRange<Value, OperandRange>(op.getInput(), resource_arg_uses); new_operands.insert(new_operands.begin(), op.getOperand(0)); func::FuncOp first_func = branches.front(); auto new_op = builder.create<CaseOrIfOp>( op.getLoc(), first_func.getFunctionType().getResults(), new_operands, op->getAttrs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
// Get new operand and result types. auto new_operands = llvm::to_vector<4>(while_op->getOperands()); auto new_result_types = llvm::to_vector<4>(while_op->getResultTypes()); Value const_output = const_wrapper.getOutputs()[0]; for (int i = 0; i < num_resource_inputs; ++i) { new_operands.push_back(const_output); new_result_types.push_back(const_output.getType()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
builder.setInsertionPoint(call); auto new_operands = llvm::to_vector(call.getArgOperands()); new_operands.push_back(token); auto new_result_types = llvm::to_vector(call.getResultTypes()); new_result_types.push_back(token.getType()); auto new_call = builder.create<func::CallOp>( call.getLoc(), new_result_types, new_symbol ? *new_symbol : call.getCallee(), new_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) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.h
SmallVector<Value> CloneOpWithReplacedOperands(OpBuilder& builder, Operation* op, ArrayRef<Value> new_operands); // Tries casting `op` with a concrete op type `T`. If the cast fails or `op` is // a `nullptr`, returns `failure` and prints a debugging message identifying // the cast attempt as `name`. template <typename T>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/jit/deadness_analysis.cc
std::vector<Predicate*> new_operands; absl::c_transform(pred->GetOperands(), std::back_inserter(new_operands), [&](Predicate* p) { return MakeNotPredicate(p); }); return kind == Predicate::Kind::kOr ? MakeAndPredicate(new_operands) : MakeOrPredicate(new_operands); } return nullptr; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 60.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
} } // The While ops operands and result types need to match SmallVector<Value, 4> new_operands; SmallVector<Value, 4> new_body_yield; SmallVector<bool, 4> removed_operand(while_op.getNumOperands(), false); llvm::SmallVector<Type, 4> types; new_operands.reserve(while_op.getNumOperands()); new_body_yield.reserve(while_op.getNumOperands());
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/transforms/decompose_hybrid_quantization.cc
Location loc = op->getLoc(); 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));
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/tensorflow/transforms/functional_control_flow_to_cfg.cc
func::FuncOp fn, OpBuilder* builder) { FunctionType fn_type = fn.getFunctionType(); llvm::SmallVector<Value, 4> operands; int num_operands = fn_type.getNumInputs(); operands.reserve(num_operands); for (int i = 0; i < num_operands; ++i) { Value val = get_arg(i); Type expected = fn_type.getInput(i); if (val.getType() != expected) { val =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 13 11:42:59 UTC 2023 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_to_vhlo.cc
for (int i = 0; i < op->getNumOperands(); ++i) { auto operand = op->getOperand(i); auto new_operand = converter.materializeArgumentConversion( rewriter, op->getLoc(), converter.convertType(operand.getType()), {operand}); op->setOperand(i, new_operand); } } // vhlo.op %1 : vhlo.tensor<...> // ==> // vhlo.op %1 : tensor<...> //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 19:48:51 UTC 2024 - 12.3K bytes - Viewed (0)