- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for GetFunctionBody (0.19 sec)
-
tensorflow/compiler/jit/compilability_check_util.cc
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( StackFrameView{node->name(), function.name(), node->GetStackTrace()});
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/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
xla_literal->element_count()); TF_ASSIGN_OR_RETURN(auto attrs, tensorflow::ConvertTensor(tensor, &builder)); builder.setInsertionPoint( &func_op.getFunctionBody().getBlocks().front().front()); // Use mhlo.Constant when it is consumed by the lowering passes since they // can't lower tf.Const. Value cst; if (use_mhlo_const) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K 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)); // We could possibly use something cheaper than
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/compiler/jit/extract_outside_compilation_pass.cc
auto cleanup_handle = gtl::MakeCleanup([&]() { auto s = flr->ReleaseHandle(handle); if (!s.ok()) { ret_status.Update(s); } }); const FunctionBody* fbody = flr->GetFunctionBody(handle); // Check if we have outside compilation nodes. *has_outside_compilation = false; for (Node* n : fbody->graph->nodes()) { if (HasNodeAttr(n->def(), outside_compilation_attr_name)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0)