- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 102 for function_ref (0.2 sec)
-
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/tensorflow/analysis/side_effect_analysis.h
Operation* op, llvm::function_ref<bool(Operation*)> filter) const; // pass the filter (returning true) will be included. const llvm::SmallVector<Operation*, 4>& DirectControlSuccessors( Operation* op) const; llvm::SmallVector<Operation*, 4> DirectControlSuccessors( Operation* op, llvm::function_ref<bool(Operation*)> filter) const;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
// Returns a failure if a cycle prevents the merge from happening correctly // without breaking dominance. The IR is left in invalid state in case of // failure. void CollectCandidateIslands( llvm::function_ref<bool(llvm::StringRef, Operation*)> is_op_calling_func_for_cluster, Operation* op, StringRef cluster_name, SmallPtrSet<Operation*, 16>& islands_set, SmallPtrSet<Operation*, 16>& wrapped_ops) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K 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) -
tensorflow/compiler/mlir/tf2xla/api/v2/cluster_tf.cc
using mlir::func::FuncOp; // Run the TF XLA Bridge based on the input pipeline, which can be either TPU // bridge pipeline or non TPU bridge pipeline. tensorflow::Status RunTFXLABridge( ModuleOp module, llvm::function_ref<void(OpPassManager &pm)> pipeline_builder, llvm::StringRef module_name = llvm::StringRef(), llvm::StringRef dump_prefix = "tf_xla_bridge_v2") { // Explicitly check that the TensorFlow dialect can constant fold ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:25:18 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/cluster_tf.cc
} // Run the TF XLA Bridge based on the input pipeline, which can be either TPU // bridge pipeline or non TPU bridge pipeline. tensorflow::Status RunTFXLABridge( ModuleOp module, llvm::function_ref<void(OpPassManager &pm)> pipeline_builder, llvm::StringRef module_name = llvm::StringRef(), llvm::StringRef dump_prefix = "tf_xla_bridge_v1") { // Explicitly check that the TensorFlow dialect can constant fold ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:25:18 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_island_coarsening.cc
// Returns a list of all of the mergable islands found in the graph. iterator_range< llvm::filter_iterator<SmallVector<MergedIsland>::const_iterator, function_ref<bool(const MergedIsland&)>>> GetMergableIslands() const { function_ref<bool(const MergedIsland&)> filter_fn = [](const MergedIsland& merged_island) { return merged_island.islands.size() > 1; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
// size variables before finally changing the function type. void ModifyFunctionSignature( func::FuncOp func, llvm::SmallDenseMap<Value, Value>* stack_var_to_size_var, llvm::function_ref<std::optional<Type>(int64_t)> arg_to_stack_type, llvm::function_ref<void(ArrayRef<BlockArgument>)> handle_new_size_vars = nullptr) { auto new_input_types = llvm::to_vector<8>(func.getFunctionType().getInputs());
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/cluster_ops_by_policy.h
// Inserts constraints for multiple values. void Insert(ValueRange value, ValueConstraint constraint); // Walk all the constraints owned by this set. void Walk(llvm::function_ref<void(Value, ValueConstraint)> walk) const; // Returns the constraint of the value if it exists, or None otherwise. std::optional<ValueConstraint> GetConstraint(Value value) const; bool HasConstraint(Value value) const;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 03:47:00 UTC 2023 - 12.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
*buffer = broadcast.getOutput(); return success(); } std::optional<RankedTensorType> GetElementTypeFromAccess( Value collection, ModuleOp module, llvm::function_ref<std::optional<Type>(Operation*)> infer_from_op) { for (auto& use : collection.getUses()) { if (auto while_op = llvm::dyn_cast<TF::WhileOp>(use.getOwner())) { auto body = while_op.body_function();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0)