- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 17 for FunctionBody (0.51 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/extract_outside_compilation_pass.cc
!HasLiftedArgs(*else_branch_function_def)) { return absl::OkStatus(); } std::unique_ptr<FunctionBody> then_branch_function_body; TF_RETURN_IF_ERROR(FunctionDefToBodyHelper( *then_branch_function_def, AttrSlice(&then_branch_func.attr()), fld, &then_branch_function_body)); std::unique_ptr<FunctionBody> else_branch_function_body; TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
for (const std::string& func_name : function_library->ReachableDefinitions(graph_def).ListFunctionNames()) { const FunctionDef* func_def = function_library->Find(func_name); std::unique_ptr<FunctionBody> func_body; absl::Status status = FunctionDefToBodyHelper( *func_def, AttrSlice(&func_def->attr()), function_library, &func_body); // This is not expected to happen in practice if (!status.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/jit/force_xla_constants_on_host_pass.cc
FunctionLibraryRuntime* flr = pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); for (Node* node : graph->nodes()) { if (CanCreateXlaKernel(node->def())) { const FunctionBody* fbody = nullptr; std::vector<int> constant_arg_indices; std::vector<int> resource_arg_indices; NameAttrList function; TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(node->def(), &function));
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/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/xla_kernel_creator.cc
XlaOpRegistry::RegisterCompilationKernels(); // Get function body, constant args, and resource args. NameAttrList function; TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(node_def, &function)); const FunctionBody* fbody = nullptr; std::vector<int> constant_arg_indices; std::vector<int> resource_arg_indices; TF_RETURN_IF_ERROR(GetBodyAndConstantsAndResources(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 22:24:01 UTC 2023 - 4.4K 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/tensorflow/translate/import_model.h
// [Experimental] // Given a Function, returns a MLIR module containing the graph, expressed with // tf_executor dialect. absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> ConvertFunctionToMlir( const FunctionBody* fbody, const FunctionLibraryDefinition& flib_def, mlir::MLIRContext* context); // Given a SavedModel, returns a MLIR module containing the functions, expressed // with tf_executor dialect.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 6.8K bytes - Viewed (0)