- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 102 for function_ref (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/mlprogram_util.cc
[](mlir::OpPassManager& pm, llvm::StringRef options, llvm::function_ref<mlir::LogicalResult(const llvm::Twine&)> errorHandler) { tensorflow::PopulateLowerToMlProgramAndHloPipeline(pm); return mlir::success(); }, [](llvm::function_ref<void(const mlir::detail::PassOptions&)>) {}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 22:13:50 UTC 2024 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sink_constant.cc
<< "\n in " << use->get() << "\n"); }); }); } private: llvm::function_ref<bool(tf_device::ClusterOp, ElementsAttr)> filter_; }; } // anonymous namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateClusterConstantSinkingPass( llvm::function_ref<bool(tf_device::ClusterOp, ElementsAttr)> filter) { return std::make_unique<ClusterConstantSinkingPass>(filter); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_util.h
bool IsArgConsumedByFallback(mlir::func::FuncOp func, int arg_index); void ForEachArgConsumedByFallback( mlir::func::FuncOp func, llvm::function_ref<void(int arg_index)> action); void ForEachArgConsumedByFallback( mlir::ModuleOp module, llvm::function_ref<void(llvm::StringRef func_name, int arg_index)> action); } // namespace fallback_async } // namespace tfrt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 22 14:25:57 UTC 2022 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_util.cc
} void ForEachArgConsumedByFallback( mlir::func::FuncOp func, llvm::function_ref<void(int arg_index)> action) { for (int arg_index = 0; arg_index < func.getNumArguments(); ++arg_index) { if (IsArgConsumedByFallback(func, arg_index)) action(arg_index); } } void ForEachArgConsumedByFallback( mlir::ModuleOp module, llvm::function_ref<void(llvm::StringRef func_name, int arg_index)> action) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 30 02:12:49 UTC 2022 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_types.h
} static Derived getChecked(function_ref<InFlightDiagnostic()> emitError, MLIRContext* context, ArrayRef<StringAttr> attrs) { return Base::getChecked(emitError, context, attrs); } static Derived get(MLIRContext* context) { return get({}, context); } // TODO(fengliuai): fix the implementation static LogicalResult verify(function_ref<InFlightDiagnostic()> emitError,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 05 07:17:01 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/saved_model/saved_model.h
// corresponding signature metadata populated in TFRTSavedModelSignatureInfo for // the function. Status MapFunctionSignaturesFromTFSavedModelMLIR( mlir::ModuleOp module, llvm::function_ref<void(const TFRTSavedModelSignatureInfo&)> map_fn); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 27 17:42:41 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/visitor.h
// and returns a `mlir::WalkResult::interrupt()` when it encounters a call whose // callee cannot be resolved to `mlir::func::FuncOp`. mlir::WalkResult WalkReachableFunctions( mlir::func::FuncOp func, llvm::function_ref<mlir::WalkResult(mlir::func::FuncOp)> callback, mlir::SymbolTableCollection* symbol_table = nullptr); // Creates a new MLIR module that contains only the given functions and all // reachable functions from them.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 03:46:51 UTC 2023 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/topological_sort.h
// must also be in extra_dependencies(op1, false). // This function is called multiple times during the topological sort, // so the implementation should preferably be constant-time. typedef llvm::function_ref<llvm::SmallVector<Operation *, 4> const &( Operation *, bool incoming)> ExtraDependenciesFunction; // Convenience function if there are no extra dependencies to declare.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 08 17:01:11 UTC 2022 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/visitor.cc
#include "mlir/IR/Visitors.h" // from @llvm-project #include "mlir/Support/LogicalResult.h" // from @llvm-project namespace mlir { namespace TF { WalkResult WalkReachableFunctions( func::FuncOp func, llvm::function_ref<WalkResult(func::FuncOp)> callback, SymbolTableCollection* symbol_table) { llvm::SmallDenseSet<Operation*> visited; llvm::SmallVector<func::FuncOp> stack; stack.push_back(func);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 03:46:51 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.h
// `infer_from_user` should check if the provided op is an accessing op that // could be used to infer the type. std::optional<RankedTensorType> GetElementTypeFromAccess( Value collection, ModuleOp module, llvm::function_ref<std::optional<Type>(Operation*)> infer_from_op); // Creates a ReadVariableOp on a local variable. Value ReadLocalVariable(Value local_var, OpBuilder builder, Location loc);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 4.8K bytes - Viewed (0)