- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for XlaCompilationResult (0.39 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h
llvm::MutableArrayRef<std::unique_ptr<mlir::Pass>> custom_legalization_passes); // Apply shape, description, and resource information to inputs and outputs // in the XlaCompilationResult. This should be called after // compilation_result->computation was set. ABSL_DEPRECATED("Not meant to be used directly and should be a util.") Status PopulateResultIOInfo(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
llvm::cl::init(false)); } // namespace namespace tensorflow { namespace { mlir::LogicalResult PrintHloModuleText( const XlaCompilationResult& compilation_result, llvm::raw_ostream& output) { const xla::HloModuleConfig module_config( compilation_result.computation->GetProgramShape().value()); auto status_or_hlo_module = xla::HloModule::CreateFromProto(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph_test.cc
return mlir_to_hlo_args; } class CompileTFGraphTest : public ::testing::Test { public: absl::StatusOr<XlaCompilationResult> CompileWithComputation( const std::variant<tpu::MlirToHloArgs, tpu::FunctionToHloArgs> computation) { XlaCompilationResult compilation_result; se::Platform* platform = se::PlatformManager::PlatformWithName(kPlatformName).value(); auto client =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler.h
// compilation (HLO) using the XlaCompiler and compiles the resulting // XlaCompilationResult into an `ExecutableType` (eg. xla::LocalExecutable) by // calling `ClientType` (eg. xla::LocalClient). // // Caches the compiled XlaCompilationResult and Executable using a // DeviceCompilationCache. Compilation is done only when there's a cache miss. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 22.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
if (VLOG_IS_ON(2)) tensorflow::DumpMlirOpToFile("build_hlo_tf_after", module_op); return absl::OkStatus(); } Status PopulateCollectiveInfo(mlir::ModuleOp module_op, XlaCompilationResult* compilation_result) { // The StringRef cast is necessary before cxx14. mlir::IntegerAttr group_key_attr = module_op->getAttrOfType<mlir::IntegerAttr>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
// Get-or-create thread pool for a given collective. static thread::ThreadPool* GetOrCreateThreadPoolForCollective( const XlaCompilationResult::CollectiveInfo& collective_info) { static absl::Mutex m(absl::kConstInit); static auto& thread_pool_cache ABSL_GUARDED_BY(m) = *new absl::node_hash_map<XlaCompilationResult::CollectiveInfo, thread::ThreadPool>(); absl::MutexLock l(&m);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0)