- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 102 for function_ref (0.25 sec)
-
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.cc
if (it == sorted_control_predecessors_.end()) return empty_operation_set_; return it->second; } llvm::SmallVector<Operation*, 4> SideEffectAnalysisInfo::DirectControlPredecessors( Operation* op, llvm::function_ref<bool(Operation*)> filter) const { llvm::SmallVector<Operation*, 4> result; auto it = sorted_control_predecessors_.find(op); if (it == sorted_control_predecessors_.end()) return result;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 41.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
// and sizes. void ModifyFunctionSignature( func::FuncOp func, Type size_type, llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, llvm::function_ref<std::optional<Type>(int64_t)> arg_to_buffer_type, llvm::function_ref<bool(int64_t)> arg_buffer_size_is_fixed) { auto new_input_types = llvm::to_vector<8>(func.getFunctionType().getInputs()); int64_t original_arg_count = new_input_types.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
// adding required gradient arguments. void ChangeFunctionInputSignature( func::FuncOp func, const llvm::SmallDenseMap<int64_t, llvm::SmallVector<string, 4>>& grads, llvm::function_ref<Type(int64_t)> ta_arg_buffer_type, llvm::function_ref<bool(int64_t)> ta_accumulate_on_write, llvm::SmallDenseMap<Value, TensorArrayStats>* stats) { int64_t original_args = func.getNumArguments();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
/// `intCalculate` is chosen to conduct the calculate. Attribute ConstFoldBinaryOp( Type result_type, ArrayRef<Attribute> operands, llvm::function_ref<APFloat(APFloat, APFloat)> float_calculate, llvm::function_ref<APInt(APInt, APInt)> int_calculate) { // Note: All types are wrapped in tensor types in TFlite. E.g., f32 is // represented as tensor<f32>. So we are only handling tensor types here.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc
// Update existing constraint with a new one. auto merged = Merge(constraint, persisted); return {constraints_[value] = merged, merged != persisted}; } void ValuesConstraintSet::Walk( llvm::function_ref<void(Value, ValueConstraint)> walk) const { for (auto &kv : constraints_) walk(kv.getFirst(), kv.getSecond()); } std::optional<ValueConstraint> ValuesConstraintSet::GetConstraint( Value value) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 27.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
for (; it != block.rend(); ++it) { CastOp cast = dyn_cast<CastOp>(*it); if (!cast || cast.getTruncate()) return std::nullopt; } return call; } using ArgMatcherFn = function_ref<bool(Value, Region&, Value, Region&)>; // Returns whether the arguments of the given 2 calls are match (after looking // through cast ops). `matcher` is the predicate used to check if two arguments // match.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
// performed in order to limit the number of values implicitly captured in this // region before outlining. std::unique_ptr<OperationPass<func::FuncOp>> CreateClusterConstantSinkingPass( llvm::function_ref<bool(tf_device::ClusterOp, ElementsAttr)> filter = {}); // Creates a pass that outlines regions of tf_device.cluster operations. std::unique_ptr<OperationPass<ModuleOp>> CreateClusterOutliningPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
// that handles the updated value for an resource argument. LogicalResult LiftArgRetResourcesForFunction( func::FuncOp func_op, const llvm::SmallDenseMap<int64_t, Type>& resource_data_types, llvm::function_ref<void(int64_t, Value)> handle_updated_arg_value) { RegionResourceHoister hoister(func_op); if (failed(hoister.Analyze())) return failure(); // Each of these resources could be read or written in the function. If its
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
return success(); } static LogicalResult VerifyCaseOrIfOpBranchFunctions( SymbolTableCollection& symbol_table, Operation* op, ArrayRef<Attribute> branches, llvm::function_ref<std::string(unsigned branch_index)> branch_name) { SmallVector<FunctionType, 2> branch_types; branch_types.reserve(branches.size()); if (llvm::any_of(op->getOperands(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/saved_model_utils.cc
return result; } gtl::FlatMap<StringPiece, const tensorflow::FunctionDef*, StringPieceHasher> FunctionNameToFunctionDefMap(const FunctionDefLibrary& library) { gtl::FlatMap<StringPiece, const tensorflow::FunctionDef*, StringPieceHasher> result; for (const FunctionDef& function_def : library.function()) { result[function_def.signature().name()] = &function_def; } return result; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 12 19:17:46 UTC 2023 - 24K bytes - Viewed (0)