- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 19 for new_operands (0.33 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
new_types.back() = rewriter.getType<tf_executor::ControlType>(); llvm::SmallVector<Value, 2> new_operands; FilterOutBlockArgControlDep(operands, new_operands); rewriter.replaceOpWithNewOp<tf_executor::ControlTriggerOp>( op, new_types, new_operands, op->getAttrs()); return success(); } }; class ConvertEnterOp : public ConversionPattern { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
auto block = b.createBlock(®ion); SmallVector<Value, 4> new_operands; new_operands.reserve(types.size()); for (Type t : llvm::ArrayRef(types).drop_back(extern_values.size())) new_operands.push_back(block->addArgument(t, loc)); for (Value v : extern_values) new_operands.push_back(v); auto call = b.create<func::CallOp>(loc, func, new_operands); b.create<YieldOp>(loc, call.getResults()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc
op.erase(); } body.eraseArguments(erase_indices); cond.eraseArguments(erase_indices); llvm::SmallVector<Value> new_operands; llvm::SmallVector<Type> new_result_types; new_operands.reserve(n - erase_indices.size()); new_result_types.reserve(n - erase_indices.size()); // After reducing, the number of results is decreased. The i-th result of old
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
llvm::SmallVector<Value, 4> new_operands; for (auto arg : llvm::enumerate(compose_func_type.getInputs())) { if (auto tensor_type = mlir::dyn_cast<TFRTensorType>(arg.value())) { auto casted = builder.create<CastOp>(op->getLoc(), tensor_type, op->getOperand(arg.index())); new_operands.push_back(casted); } else if (auto list_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc
// is qint and its defining op is not already an int->qint CastOp. llvm::SmallVector<Value, 4> new_operands; for (int i = 0; i < operands.size(); ++i) { Type orig_op_type = op->getOperandTypes()[i]; if (IsIllegalType(orig_op_type) && !IsQintValueDefinedByIntToQintCast(op->getOperand(i))) { new_operands.push_back(rewriter.create<TF::CastOp>( op->getLoc(), orig_op_type, operands[i]));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
return failure(); } assert(arg_it->second == new_operands.size()); new_operands.push_back(it->getSecond().size); } OpBuilder builder(call); auto new_call = builder.create<CallOp>( call.getLoc(), info.decomposed_callee.getFunctionType().getResults(), new_operands, call->getAttrs()); new_call->setAttr( "f", SymbolRefAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
llvm::SmallVector<Type, 4> new_operand_types; llvm::SmallVector<Value> new_operands; auto operands = while_op->getOperands(); const int num_operands = while_op->getNumOperands(); llvm::BitVector skip_indices(num_operands); for (int i : arguments_to_erase) skip_indices.set(i); for (int i = 0; i < num_operands; ++i) { if (!skip_indices.test(i)) { new_operand_types.emplace_back(argument_types[i]);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
return failure(); } assert(arg_it->second == new_operands.size()); new_operands.push_back(it->getSecond()); } OpBuilder builder(call); auto new_call = builder.create<CallOp>( call.getLoc(), info.decomposed_callee.getFunctionType().getResults(), new_operands, call->getAttrs()); new_call->setAttr( "f", SymbolRefAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K 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)