- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 41 for function_ref (0.21 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/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) -
tensorflow/compiler/mlir/tfrt/saved_model/saved_model.cc
return std::make_pair(dtype, tensorflow::PartialTensorShape(dims)); } } // namespace Status MapFunctionSignaturesFromTFSavedModelMLIR( mlir::ModuleOp module, llvm::function_ref<void(const TFRTSavedModelSignatureInfo&)> map_fn) { // Create bound inputs for each functions. mlir::SymbolTable symbol_table(module); tensorflow::Status status = absl::OkStatus();
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/experimental/tac/transforms/device_transform.cc
}; auto dequant_values = mlir::cast<DenseIntOrFPElementsAttr>(input_values) .mapValues(FloatType::getF32(rewriter.getContext()), llvm::function_ref<DequantizeFuncType>(dequantize_func)); rewriter.replaceOpWithNewOp<TFL::ConstOp>(dequant_op, dequant_op.getType(), dequant_values); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.4K bytes - Viewed (0)