- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 17 for function_ref (0.33 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/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) -
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/v2/tf_executor_to_graph.h
Status ConvertMlirFunctionToFunctionLibraryDef(mlir::func::FuncOp func, const GraphExportConfig& configs, FunctionDef* function_def); } // namespace v2 } // namespace tf2xla } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.h
Status ConvertMlirFunctionToFunctionLibraryDef(mlir::func::FuncOp func, const GraphExportConfig& configs, FunctionDef* function_def); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.h
// Decomposes the op in the NodeDef to a set of primitive ops according to the // decompose library in the context. Wrap the decomposed result in a // FunctionDef. absl::StatusOr<FunctionDef> ExpandNode(const NodeDef& node_def, StringPiece func_name); // Runs the decompose passes on the user_module. Status DecomposeGraph(mlir::ModuleOp user_module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 11:12:54 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/c/eager/graph_function.h
namespace graph { using tensorflow::AbstractFunction; // Thin wrapper around a FunctionDef. class GraphFunction : public AbstractFunction { public: explicit GraphFunction(FunctionDef fdef); ~GraphFunction() override; // GraphFunction maybe stay alive for the duration of the returned // FunctionDef. Status GetFunctionDef(const FunctionDef** fdef) override; // Returns a shared reference to the wrapped function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 04 19:49:06 UTC 2024 - 1.8K bytes - Viewed (0) -
tensorflow/c/eager/abstract_function.h
// Returns which subclass is this instance of. AbstractFunctionKind getKind() const { return kind_; } // Returns the AbstractFunction as a FunctionDef. virtual Status GetFunctionDef(const FunctionDef**) = 0; // Returns a shared reference to the wrapped function. virtual absl::StatusOr<core::RefCountPtr<FunctionRecord>> GetFunctionRecord() = 0; private: const AbstractFunctionKind kind_; }; using AbstractFunctionPtr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 04 19:49:06 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/c/eager/graph_function.cc
namespace tensorflow { namespace tracing { namespace graph { GraphFunction::GraphFunction(FunctionDef fdef) : AbstractFunction(kGraph), func_record_(new FunctionRecord(std::move(fdef), {}, true)) {} GraphFunction::~GraphFunction() {} Status GraphFunction::GetFunctionDef(const FunctionDef **fdef) { *fdef = &(func_record_->fdef()); return absl::OkStatus(); } } // namespace graph
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 04 19:49:06 UTC 2024 - 1.3K bytes - Viewed (0)