- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for GetFunctionLibraryDefinition (0.47 sec)
-
tensorflow/compiler/jit/xla_compiler_options_util.cc
if (stream != nullptr) { options.device_ordinal = stream->parent()->device_ordinal(); } options.device_type = xla_device_compiler.device_type(); options.flib_def = function_library.GetFunctionLibraryDefinition(); options.graph_def_version = function_library.graph_def_version(); options.allow_cpu_custom_calls = (platform_info.platform_id() == se::host::kHostPlatformId);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/saved_model/saved_model_test.cc
// The module contains an XLA function, as well as a while body and a while // condition within the XLA function. EXPECT_EQ(fallback_state->process_function_library_runtime() .GetFunctionLibraryDefinition() ->num_functions(), 3); } TEST(SavedModelTest, ConvertTfMlirToBefExportingXlaReduceWindow) { std::string saved_model_mlir_path = tensorflow::GetDataDependencyFilepath(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 13 01:17:29 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
std::shared_ptr<const tensorflow::NodeProperties> props; absl::Status status = tensorflow::NodeProperties::CreateFromNodeDef( *nodedef_or.value(), params_.function_library->GetFunctionLibraryDefinition(), &props); if (!status.ok()) { return op_->emitRemark() << "failed to create NodeProperties: " << status.ToString(); } tensorflow::OpKernel* op_kernel_raw;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
options = GenerateCompilerOptionsForTfrtTpu(*xla_device_compiler, *flr); } else { options.device_type = compilation_device_type; options.flib_def = flr->GetFunctionLibraryDefinition(); options.graph_def_version = flr->graph_def_version(); options.allow_cpu_custom_calls = (platform_info.platform_id() == se::host::kHostPlatformId);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cluster_util.cc
CallTargetListTy GetCallTargetListFromNode( const Node& n, FunctionLibraryRuntime* lib_runtime) { const FunctionLibraryDefinition& flib_def = *lib_runtime->GetFunctionLibraryDefinition(); if (flib_def.Find(n.type_string())) { NameAttrList callee; callee.set_name(n.type_string()); *callee.mutable_attr() = n.def().attr(); return {callee}; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/c/kernels.cc
TF_Status* status) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); auto fdef_lib = cc_ctx->function_library()->GetFunctionLibraryDefinition()->ToProto(); auto cc_status = tensorflow::MessageToBuffer(fdef_lib, serialized_function_def_library); tensorflow::Set_TF_Status_from_Status(status, cc_status); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
LogNotCompilable(node, uncompilable_reason); return false; } string uncompilable_reason; if (IsFunctionCall(*lib_runtime->GetFunctionLibraryDefinition(), node)) { if (!IsCompilableCall(node.def(), lib_runtime, stack_trace, encapsulating_function, uncompilable_nodes)) { LogNotCompilable(node, "unsupported function");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0)