- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for function_ref (0.18 sec)
-
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/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/compiler/mlir/tensorflow/transforms/shape_inference.cc
} }; using ValuePortResultMap = absl::flat_hash_map<ValuePort, Attribute, ValuePortHasher>; using ComputedQueryFn = function_ref<bool(ValuePort)>; using ValueQueryFn = function_ref<Attribute(const ValuePort&)>; using ValuePortInputs = SmallVectorImpl<ValuePort>; // Note: Following implements the rank 1 pack op case so could be // generalized. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
n->DebugString()); } } return absl::OkStatus(); } // Cheap check to tell whether FunctionDef contains a lifted argument. bool HasLiftedArgs(const FunctionDef& function_def) { return absl::c_any_of(function_def.node_def(), [](const NodeDef& node_def) { return (node_def.op() == "Placeholder" && node_def.attr().find(kXlaLiftedArgOutsideCompilationAttrName) !=
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
//===----------------------------------------------------------------------===// // ReshapeOp //===----------------------------------------------------------------------===// namespace { using ReshapeErrorHandler = llvm::function_ref<LogicalResult(const llvm::Twine &)>; LogicalResult GetReshapeOutputType(Value tensor, Value shape, ReshapeErrorHandler error_handler,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
const FunctionDefLibrary& actual, string* diff) { std::unordered_map<string, const FunctionDef*> actual_index; for (const FunctionDef& function : actual.function()) { actual_index[function.signature().name()] = &function; } for (const FunctionDef& expected_function : expected.function()) { auto it = actual_index.find(expected_function.signature().name());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
} Status ImporterBase::ConvertDeferredFunctions() { while (!deferred_functions_.empty()) { auto conversion_metadata = deferred_functions_.front(); deferred_functions_.pop(); const FunctionDef* func_def = graph_flib_.Find(conversion_metadata.function_name); // Converts the graph to an MLIR function and adds it to the module. // We populate the NodeSpec so that all the _Arg ops get their shape
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)