- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for FunctionBody (0.3 sec)
-
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); TF_CHECK_OK(s.ToGraph(g.get())); std::vector<std::unique_ptr<FunctionBody>> fbodies; TF_CHECK_OK(RearrangeFunctionArguments( [&](const NameAttrList &function, const FunctionBody **fbody) { std::unique_ptr<FunctionBody> new_fbody; TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(*fld.Find(function.name()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
host_compute_core, &fld, &shape_inference_graphs, &has_outside_compilation)); // Get rewritten XLA computation function. std::unique_ptr<FunctionBody> xla_fbody; TF_CHECK_OK(FunctionDefToBodyHelper(*fld.Find("cluster_rewritten"), AttrSlice(), &fld, &xla_fbody)); auto node_name_index = xla_fbody->graph->BuildNodeNameIndex();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.h
Status GetBodyAndConstantsAndResources(FunctionLibraryRuntime* flr, const NameAttrList& function, const FunctionBody** fbody, std::vector<int>* constant_arg_indices, std::vector<int>* resource_arg_indices);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
<< uncompilable_reason << " : " << s; return false; } auto release_handle_on_return = gtl::MakeCleanup( [&] { TF_CHECK_OK(lib_runtime->ReleaseHandle(handle)); }); const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); bool is_compilable = true; for (const Node* node : fbody->graph->op_nodes()) { stack_trace->emplace_back(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
TF_RETURN_IF_ERROR(graph.status()); return *graph; } } } static absl::StatusOr<std::vector<XlaCompiler::Argument>> BuildXlaCompilerArgumentFromTensorSpec( const FunctionBody* fbody, absl::Span<int const> must_be_constant_idxs, absl::Span<const Tensor* const> inputs, absl::Span<VariableInfo const> variable_args, absl::Span<const ArgShapeAndDType> flat_arg_shape_and_dtype) {
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/mlir/python/mlir.cc
if (fdef == nullptr) { s = tensorflow::errors::NotFound("Cannot find function ", function_name); tsl::Set_TF_Status_from_Status(status, s); return "// error"; } std::unique_ptr<tensorflow::FunctionBody> fbody; s = FunctionDefToBodyHelper(*fdef, tensorflow::AttrSlice(), &flib_def, &fbody); if (!s.ok()) { tsl::Set_TF_Status_from_Status(status, s); return "// error"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cluster_util.cc
return true; } auto release_handle_on_return = gtl::MakeCleanup( [&] { TF_CHECK_OK(lib_runtime->ReleaseHandle(handle)); }); const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); TF_RETURN_IF_ERROR(GetNodesRelatedToRefVariablesInDirection( *fbody->graph, lib_runtime, direction, depth + 1, &callee_ref_nodes));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 21.3K bytes - Viewed (0)