- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for getFunctionType (0.21 sec)
-
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
Value max_output_size = func_.getArgument(2); Value iou_threshold = func_.getArgument(3); Value score_threshold = func_.getArgument(4); auto output_type0 = func_.getFunctionType().getResult(0); auto output_type1 = func_.getFunctionType().getResult(1); OpBuilder builder(func_.getBody()); auto op = builder.create<mlir::TFL::NonMaxSuppressionV4Op>( func_.getLoc(), output_type0, output_type1, boxes, scores,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
<< func_.getNumArguments(); } if (func_.getFunctionType().getNumResults() != 1) { return func_.emitWarning() << "Invalid number of results from " << kImageWarping << ": " << func_.getFunctionType().getNumResults(); } // Check types and shapes. auto image_type = mlir::dyn_cast_or_null<RankedTensorType>( func_.getFunctionType().getInput(0));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc
llvm::SmallVector<Type, 4> operand_types; operand_types.reserve(live_ins.size()); for (Value v : live_ins) operand_types.emplace_back(v.getType()); auto func_type = builder->getFunctionType(operand_types, op.getResultTypes()); std::string func_name; if (auto outlined_func_name = op->template getAttrOfType<StringAttr>( TF::kClusterOutlinedFunctionNameAttr)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/utils/test_metadata_config.cc
auto main_fn = module.lookupSymbol<mlir::func::FuncOp>(kEntryFuncName); if (!main_fn) { return absl::InternalError("Could not find main function in MLIR Module."); } mlir::FunctionType func_type = main_fn.getFunctionType(); for (auto input_type : func_type.getInputs()) { tensorflow::TensorShape tensor_shape; xla::Shape xla_shape = xla::TypeToShape(input_type); TF_RETURN_IF_ERROR(tensorflow::TensorShape::BuildTensorShape(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/saved_model/saved_model.cc
// functions. auto func_names = mlir::tf_saved_model::GetExportedNames(func); if (func_names.empty()) return mlir::WalkResult::advance(); auto func_type = func.getFunctionType(); // Here we walk through each arguments and find out the input/output names, // and input devices, variables used by this function. llvm::SmallVector<llvm::StringRef, 4> input_names; llvm::SmallVector<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_jax_random.cc
} void LegalizeJaxRandomPass::runOnOperation() { auto func = getOperation(); if (!IsJaxRandomUniform(func) && !IsJaxRandomNormal(func)) return; auto result_tuple_ty = mlir::dyn_cast_or_null<TupleType>(func.getFunctionType().getResult(0)); if (!result_tuple_ty) return; if (result_tuple_ty.size() != 1) return; auto result_ty = mlir::dyn_cast<ShapedType>(result_tuple_ty.getType(0)); func.eraseBody(); func.addEntryBlock();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_functional_ops.cc
void UpdateFuncType(func::FuncOp func) { Operation* terminator = func.front().getTerminator(); auto return_types = llvm::to_vector<4>(terminator->getOperandTypes()); FunctionType func_type = func.getFunctionType(); if (llvm::ArrayRef(return_types) == func_type.getResults()) return; auto updated_type = FunctionType::get(func.getContext(), func_type.getInputs(), return_types); func.setType(updated_type); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
main_func.getArguments().front().setType(tensor_string_type); main_func.setType( FunctionType::get(builder.getContext(), {tensor_string_type}, main_func.getFunctionType().getResults())); // Name of the main function for the eventual executable needs to be set. main_func->setAttr(kTfSavedModelExportedNamesAttr, builder.getStrArrayAttr({func_name}));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
// Update the function type. func.setType(mlir::FunctionType::get(module.getContext(), func.getBody().getArgumentTypes(), func.getFunctionType().getResults())); } return success(); } } // namespace tf_saved_model
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0)