- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for ProcessFunctionLibraryRuntime (0.31 sec)
-
tensorflow/compiler/jit/get_compiler_ir.h
#include "absl/strings/string_view.h" #include "absl/types/span.h" #include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "tensorflow/core/platform/statusor.h" namespace tensorflow { class ProcessFunctionLibraryRuntime; class Device; class Tensor; class TensorHandle; class EagerContext; enum class IrExportStage { HLO, HLO_NO_METADATA, HLO_SERIALIZED, OPTIMIZED_HLO, OPTIMIZED_HLO_SERIALIZED,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/jit/force_xla_constants_on_host_pass.cc
Graph* graph = options.graph->get(); OptimizerOptions opts; auto pflr = std::make_unique<ProcessFunctionLibraryRuntime>( nullptr, options.session_options->env, /*config=*/nullptr, TF_GRAPH_DEF_VERSION, options.flib_def, opts); FunctionLibraryRuntime* flr = pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); for (Node* node : graph->nodes()) { if (CanCreateXlaKernel(node->def())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/jit/test_util.cc
FunctionDefLibrary()); if (fdef.has_value()) { TF_CHECK_OK(lib_def_->AddFunctionDef(*fdef)); } pflr_ = std::make_unique<ProcessFunctionLibraryRuntime>( device_mgr_.get(), Env::Default(), /*config=*/nullptr, TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, /*default_thread_pool=*/nullptr, /*cluster_flr=*/nullptr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.h
std::unique_ptr<tensorflow::ScopedStepContainer> step_container_; std::unique_ptr<tensorflow::FunctionLibraryDefinition> flib_def_; std::unique_ptr<tensorflow::ProcessFunctionLibraryRuntime> pflr_; tensorflow::OpKernelContext::Params params_; xla::XlaBuilder xla_builder_; }; } // namespace mhlo } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 5K bytes - Viewed (0)