| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 346 TF_ASSERT_OK(fld.AddFunctionDef(cond_fdef)); in TEST()
349 TF_ASSERT_OK(fld.AddFunctionDef(body_fdef)); in TEST()
379 TF_ASSERT_OK(fld.AddFunctionDef(cond_fdef)); in TEST()
382 TF_ASSERT_OK(fld.AddFunctionDef(body_fdef)); in TEST()
|
| D | functionalize_while.cc | 248 return library->AddFunctionDef(*lookup_library->Find(function_name)); in AddMissingFunctionByName()
286 library->AddFunctionDef(*lookup_library->Find(node.op()))); in AddMissingFunctionDef()
508 TF_RETURN_IF_ERROR(library->AddFunctionDef(cond_fdef)); in FunctionalizeLoop()
509 TF_RETURN_IF_ERROR(library->AddFunctionDef(body_fdef)); in FunctionalizeLoop()
|
| D | functionalize_control_flow.cc | 224 TF_RETURN_IF_ERROR(fld->AddFunctionDef(functionalized_fdef)); in FunctionalizeControlFlowForFunction()
|
| D | tf2xla_util.cc | 97 TF_RETURN_IF_ERROR(fld->AddFunctionDef(*fdef)); in CopyAssociatedFunctions()
207 TF_RETURN_IF_ERROR(fld->AddFunctionDef(replace_fdef)); in PropagateConstIntoFuncAttr()
|
| D | xla_compiler_test.cc | 899 TF_ASSERT_OK(flib_def_->AddFunctionDef(FillFn())); in TEST_F()
939 TF_ASSERT_OK(local_flib_def->AddFunctionDef(test::function::XTimesTwo())); in TEST_F()
1342 TF_ASSERT_OK(flib_def_->AddFunctionDef(fn)); in TEST_F()
|
| D | functionalize_cond.cc | 745 TF_RETURN_IF_ERROR(library->AddFunctionDef(body_fdef)); in BuildIfNode()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | optimization_registry_test.cc | 49 return options.flib_def->AddFunctionDef(test::function::WXPlusB()); in Run()
|
| D | graph_execution_state.cc | 710 TF_RETURN_IF_ERROR((*optimized_flib)->AddFunctionDef(fdef)); in OptimizeGraph()
|
| D | process_function_library_runtime.cc | 679 TF_RETURN_IF_ERROR(data->overlay_lib_.AddFunctionDef(shard)); in InstantiateMultiDevice()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_test.cc | 1009 TEST(FunctionLibraryDefinitionTest, AddFunctionDef) { in TEST() argument
1016 TF_EXPECT_OK(lib_def.AddFunctionDef(test::function::WXPlusB())); in TEST()
1035 Status s = lib_def.AddFunctionDef(fdef); in TEST()
1042 TF_EXPECT_OK(lib_def.AddFunctionDef(test::function::XTimesTwo())); in TEST()
1043 TF_EXPECT_OK(lib_def.AddFunctionDef(test::function::WXPlusB())); in TEST()
|
| D | function.h | 339 Status AddFunctionDef(const FunctionDef& fdef) LOCKS_EXCLUDED(mu_);
|
| D | function.cc | 1032 TF_CHECK_OK(AddFunctionDef(it.second->fdef)); in FunctionLibraryDefinition()
1074 Status FunctionLibraryDefinition::AddFunctionDef(const FunctionDef& fdef) { in AddFunctionDef() function in tensorflow::FunctionLibraryDefinition
1468 const Status added = reachable_flib.AddFunctionDef(*func); in ReachableFunctionLibraryDefinition()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | filter_fusion.cc | 121 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_predicate)); in OptimizeAndCollectStats()
|
| D | map_fusion.cc | 142 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
|
| D | map_and_filter_fusion.cc | 163 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
|
| D | rebatch.cc | 218 TF_RETURN_IF_ERROR(flib->AddFunctionDef(func_def)); in RecursivelyHandleOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 343 Status EagerContext::AddFunctionDef(const FunctionDef& fdef) { in AddFunctionDef() function in tensorflow::EagerContext
345 TF_RETURN_IF_ERROR(func_lib_def_.AddFunctionDef(fdef)); in AddFunctionDef()
|
| D | context.h | 151 Status AddFunctionDef(const FunctionDef& fdef);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 596 TF_RETURN_IF_ERROR(fld->AddFunctionDef(host_graph_fdef)); in ConstructHostGraph()
952 TF_RETURN_IF_ERROR(fld->AddFunctionDef(oc_host_graph_fdef)); in BuildHostGraphForIfNode()
1222 TF_RETURN_IF_ERROR(fld->AddFunctionDef(oc_host_graph_fdef)); in BuildHostGraphForWhileNode()
1270 TF_RETURN_IF_ERROR(fld->AddFunctionDef(oc_host_graph_fdef)); in BuildHostGraphForFuncCallNode()
1663 TF_RETURN_IF_ERROR(fld->AddFunctionDef(shape_inference_fdef)); in ExtractOutsideCompilationForFunction()
1704 TF_RETURN_IF_ERROR(fld->AddFunctionDef(updated_fdef)); in ExtractOutsideCompilationForFunction()
|
| D | extract_outside_compilation_pass_test.cc | 885 TF_CHECK_OK(fld.AddFunctionDef(*xla_fdef)); in TEST_F()
|
| D | encapsulate_subgraphs_pass.cc | 1135 TF_RETURN_IF_ERROR(library->AddFunctionDef(fdef)); in BuildFunctionDef()
1178 TF_RETURN_IF_ERROR(library->AddFunctionDef(fdef)); in AddShapeInferenceInfo()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 633 TF_RETURN_IF_ERROR(flib.AddFunctionDef(func_def)); in Optimize()
768 TF_RETURN_IF_ERROR(flib->AddFunctionDef(fdef)); in OptimizeGraph()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 272 return context->Context()->AddFunctionDef(request->function_def()); in RegisterFunction()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 847 status->status = ctx->context.AddFunctionDef(function_def); in TFE_ContextAddFunctionDef()
852 status->status = ctx->context.AddFunctionDef(function->fdef); in TFE_ContextAddFunction()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions_test.cc | 382 TF_ASSERT_OK(flib.AddFunctionDef(FunctionDefHelper::Define( in TEST_F()
|