- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for addOperands (0.44 sec)
-
tensorflow/compiler/mlir/python/mlir_wrapper/ops.cc
[](mlir::OperationState& state, std::vector<mlir::Type> tys) { state.addTypes(mlir::ArrayRef<mlir::Type>(tys)); }) .def("addOperands", [](mlir::OperationState& os, std::vector<mlir::Value> ops) { os.addOperands(mlir::ArrayRef<mlir::Value>(ops)); }) .def("addRegion", py::overload_cast<>(&mlir::OperationState::addRegion), py::return_value_policy::reference);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 30 02:12:49 UTC 2022 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/gpu_fusion.cc
// Build the newly fused operation to replace the batch norm OperationState state(batch_norm.getLoc(), _FusedBatchNormExOp::getOperationName()); state.addOperands(batch_norm.getOperands()); if (side_input) state.operands.push_back(side_input); state.addTypes(batch_norm.getResultTypes()); state.addAttributes(batch_norm->getAttrs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir_wrapper/mlir_wrapper.pyi
def getRegion(self, arg0: int) -> Region: ... def getResult(self, arg0: int) -> OpResult: ... class OperationState: def __init__(self, arg0: Location, arg1: str) -> None: ... def addOperands(self, arg0: list[Value]) -> None: ... def addRegion(self) -> Region: ... def addTypes(self, arg0: list[Type]) -> None: ... class RankedTensorType(Type): def __init__(self, *args, **kwargs) -> None: ...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 09 17:10:09 UTC 2023 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
auto attr = ReadAttr(flex_buffer_map, &builder, custom_op.getCustomCode().str()); OperationState op_state(custom_op.getLoc(), custom_op.getCustomCode().str()); op_state.addOperands(custom_op.getOperands()); llvm::SmallVector<mlir::Type, 4> output_tys; for (int i = 0; i < custom_op.getNumResults(); i++) { output_tys.push_back(custom_op.getType(i)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
const std::string tf_op_full_name = llvm::Twine("tf.", tf_op_name).str(); // Create the TF op OperationState op_state(op.getLoc(), tf_op_full_name); op_state.addOperands(op.getOperands()); op_state.addTypes(op.getResultTypes()); SmallVector<NamedAttribute, 2> attrs; std::string parsed_op_name; tensorflow::NodeDef node_def;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
mlir::OpBuilder op_builder(func.getBody()); // Create the TF op const std::string tf_op_full_name = absl::StrCat("tf.", node_def.op()); mlir::OperationState op_state(loc, tf_op_full_name); op_state.addOperands(func.getArguments()); op_state.addTypes(output_tys); op_state.addAttributes(attrs); mlir::Operation* tf_op = op_builder.create(op_state); op_builder.create<mlir::func::ReturnOp>(loc, tf_op->getResults());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
OperationState op_state(loc, op_name); for (auto input_num : op.inputs) { if (input_num == -1) { assert(optional_arg_marker != nullptr); op_state.addOperands({optional_arg_marker}); } else { op_state.addOperands({vals_map.at(input_num)}); } } for (auto output_num : op.outputs) { auto& tensor = *tensors.at(output_num);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
DCHECK(succeeded( VerifyCompatibleTypes(input.getType(), replicated_input.second))); state->addOperands(input); } block.addArgument(replicated_input.second, state->location); } for (auto packed_input : packed_inputs) { state->addOperands(packed_input); block.addArgument(packed_input.getType(), state->location); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc
// op for some other operand. Update defining_ops. if (defining_ops.insert(new_control_input.getDefiningOp()).second) { operands.push_back(new_control_input); } } state.addOperands(operands); Operation* new_op = builder.create(state); item.replaceAllUsesWith(new_op); new_op->setAttrs(item.getAttrDictionary()); item.erase(); } AddStatefulControlFlowDependencies(graph_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
[{ auto resultType = OpTrait::util::getBroadcastedType(lhs.getType(), rhs.getType()); if (!resultType) mlir::emitError($_state.location, "non-broadcastable operands"); $_state.addOperands({lhs, rhs}); $_state.types.push_back(resultType); }]>; def TFL_FusedBroadcastableBinaryBuilder : OpBuilder<(ins "Value":$lhs, "Value":$rhs, "StringAttr":$fusedActivationFunction), [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0)