Home
last modified time | relevance | path

Searched refs:AddFunctionDef (Results 1 – 25 of 53) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcontext_test.cc143 TEST_F(EagerContextTest, AddFunctionDef) { in TEST_F() argument
161 TF_EXPECT_OK(context()->AddFunctionDef(x_times_two)); in TEST_F()
182 TF_EXPECT_OK(context()->AddFunctionDef(x_times_two)); in TEST_F()
198 TF_EXPECT_OK(context()->AddFunctionDef(x_times_two_copy)); in TEST_F()
219 TF_EXPECT_OK(context()->AddFunctionDef(x_times_two)); in TEST_F()
237 Status s = context()->AddFunctionDef(x_times_two_copy); in TEST_F()
265 Status s = context()->AddFunctionDef(assert_and_identity); in TEST_F()
324 Status s = context()->AddFunctionDef(x_times_two); in TEST_F()
H A Dcontext.cc920 return AddFunctionDef(fdef, FunctionDefLibrary(), in AddFunctionDefWithStackTraces()
924 Status EagerContext::AddFunctionDef(const FunctionDef& fdef) { in AddFunctionDef() function in tensorflow::EagerContext
925 return AddFunctionDef(fdef, FunctionDefLibrary(), in AddFunctionDef()
929 Status EagerContext::AddFunctionDef(const FunctionDef& fdef, in AddFunctionDef() function in tensorflow::EagerContext
965 TF_RETURN_IF_ERROR(func_lib_def_.AddFunctionDef(fdef, stack_traces)); in AddFunctionDef()
H A Deager_operation_test.cc64 TF_ASSERT_OK(ctx->AddFunctionDef(function_def)); in TEST()
H A Dcontext.h233 Status AddFunctionDef(const FunctionDef& fdef) override;
242 Status AddFunctionDef(const FunctionDef& fdef,
H A Dcore.cc170 return AddFunctionDef(*fdef); in RegisterFunction()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfunction_test.cc1041 TF_CHECK_OK(lib_def.AddFunctionDef(test::function::XTimesTwo())); in TEST()
1049 TF_CHECK_OK(lib_def.AddFunctionDef(test::function::XTimesTwo())); in TEST()
1060 TF_CHECK_OK(lib_def.AddFunctionDef(test::function::XTimesTwo())); in TEST()
1077 TEST(FunctionLibraryDefinitionTest, AddFunctionDef) { in TEST() argument
1079 TF_CHECK_OK(lib_def.AddFunctionDef(test::function::XTimesTwo())); in TEST()
1091 Status s = lib_def.AddFunctionDef(fdef); in TEST()
1098 TF_EXPECT_OK(lib_def.AddFunctionDef(test::function::XTimesTwo())); in TEST()
1126 TF_CHECK_OK(lib_def.AddFunctionDef(test::function::XTimesTwo())); in TEST()
1140 TF_CHECK_OK(lib_def.AddFunctionDef(test::function::XTimesTwo())); in TEST()
1141 TF_CHECK_OK(lib_def.AddFunctionDef(test::function::XAddX())); in TEST()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dtf2xla_util_test.cc351 TF_ASSERT_OK(fld.AddFunctionDef(cond_fdef)); in TEST()
354 TF_ASSERT_OK(fld.AddFunctionDef(body_fdef)); in TEST()
384 TF_ASSERT_OK(fld.AddFunctionDef(cond_fdef)); in TEST()
387 TF_ASSERT_OK(fld.AddFunctionDef(body_fdef)); in TEST()
437 TF_ASSERT_OK(fld.AddFunctionDef(fdef)); in TEST()
451 TF_ASSERT_OK(fld.AddFunctionDef(fdef)); in TEST()
466 TF_ASSERT_OK(fld.AddFunctionDef(fdef)); in TEST()
H A Dtf2xla_util.cc101 TF_RETURN_IF_ERROR(fld->AddFunctionDef(*fdef)); in CopyAssociatedFunctions()
261 TF_RETURN_IF_ERROR(fld->AddFunctionDef( in PropagateConstIntoFuncAttr()
270 TF_RETURN_IF_ERROR(fld->AddFunctionDef( in PropagateConstIntoFuncAttr()
1109 TF_RETURN_IF_ERROR(fld->AddFunctionDef(new_fdef)); in RewriteTensorListWithConstElement()
H A Dxla_compiler_test.cc970 TF_ASSERT_OK(flib_def_->AddFunctionDef(FillFn())); in TEST_F()
1009 TF_ASSERT_OK(local_flib_def->AddFunctionDef(test::function::XTimesTwo())); in TEST_F()
1054 TF_ASSERT_OK(flib_def_->AddFunctionDef(SliceFn())); in TEST_F()
1490 TF_ASSERT_OK(flib_def_->AddFunctionDef(fn)); in TEST_F()
1661 TF_ASSERT_OK(flib_def.AddFunctionDef(fdef)); in TEST_F()
1672 TF_ASSERT_OK(flib_def.AddFunctionDef(fdef)); in TEST_F()
1735 TF_ASSERT_OK(flib_def.AddFunctionDef(fdef)); in TEST_F()
1754 TF_ASSERT_OK(flib_def.AddFunctionDef(fdef)); in TEST_F()
H A Drearrange_function_argument.cc370 fld->AddFunctionDef(new_fdef, fld->GetStackTraces(attr_value.name()))); in MaybeRewriteWhileNode()
464 TF_RETURN_IF_ERROR(fld->AddFunctionDef(new_fdef, stack_traces)); in MaybeRewriteIfNode()
H A Dfunctionalize_while.cc429 TF_RETURN_IF_ERROR(library->AddFunctionDef(cond_fdef)); in FunctionalizeLoop()
430 TF_RETURN_IF_ERROR(library->AddFunctionDef(body_fdef)); in FunctionalizeLoop()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcontrol_flow_deps_to_chains.cc309 flib_def->AddFunctionDef(modified_body, in Run()
313 flib_def->AddFunctionDef(modified_cond, in Run()
323 graph_lib->AddFunctionDef(modified_body, in Run()
327 graph_lib->AddFunctionDef(modified_cond, in Run()
H A Dinline_function_utils_test.cc53 TF_ASSERT_OK(flib_def.AddFunctionDef(fdef)); in TEST()
H A Doptimization_registry_test.cc54 return options.flib_def->AddFunctionDef(test::function::WXPlusB()); in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dextract_outside_compilation_pass.cc571 TF_RETURN_IF_ERROR(fld->AddFunctionDef(function_def)); in AddFunctionWithNewName()
889 TF_RETURN_IF_ERROR(fld->AddFunctionDef(rewritten_fdef)); in PostprocessLiftedArgsForCall()
1478 TF_RETURN_IF_ERROR(fld->AddFunctionDef(oc_host_graph_fdef)); in BuildHostGraphForIfNode()
1551 TF_RETURN_IF_ERROR(fld->AddFunctionDef(replace_fdef)); in AddSendLoopPredToLoopCond()
1753 TF_RETURN_IF_ERROR(fld->AddFunctionDef(oc_host_graph_fdef)); in BuildHostGraphForWhileNode()
1802 TF_RETURN_IF_ERROR(fld->AddFunctionDef(oc_host_graph_fdef)); in BuildHostGraphForFuncCallNode()
1983 TF_RETURN_IF_ERROR(fld->AddFunctionDef(then_branch_host_fdef)); in ExtractOutsideCompilationForIfNode()
2000 TF_RETURN_IF_ERROR(fld->AddFunctionDef(else_branch_host_fdef)); in ExtractOutsideCompilationForIfNode()
2085 TF_RETURN_IF_ERROR(fld->AddFunctionDef(cond_host_fdef)); in ExtractOutsideCompilationForWhileNode()
2101 TF_RETURN_IF_ERROR(fld->AddFunctionDef(body_host_fdef)); in ExtractOutsideCompilationForWhileNode()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dfallback_test_util.cc48 tensorflow::Status status = eager_context->AddFunctionDef(dummy_function_def); in CreateFallbackTestExecutionContext()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
H A Dnode_expansion_pass.cc74 TF_RETURN_IF_ERROR(ctx.AddFunctionDef(func)); in Run()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
H A Dflat_tensor_function.cc52 TF_RETURN_IF_ERROR(ctx->AddFunctionDef(*function_def)); in Create()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dgraph_partition.cc286 TF_RETURN_IF_ERROR(graph->mutable_flib_def()->AddFunctionDef(new_fdef)); in BuildPartitionedCallOp()
376 TF_RETURN_IF_ERROR(graph->mutable_flib_def()->AddFunctionDef(id_fdef)); in BuildStatefulPartitionedCallOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dfilter_fusion.cc121 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_predicate)); in OptimizeAndCollectStats()
H A Dmap_fusion.cc141 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
H A Dmeta_optimizer.cc143 TF_RETURN_IF_ERROR(flib.AddFunctionDef(func_def)); in Optimize()
H A Dmap_and_filter_fusion.cc235 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
/aosp_15_r20/external/tensorflow/tensorflow/core/function/
H A Druntime_client.cc101 return this->eager_ctx_.AddFunctionDef(fdef); in CreateFunction()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dimmediate_execution_context.h120 virtual Status AddFunctionDef(const FunctionDef& fdef) = 0;

123