Home
last modified time | relevance | path

Searched refs:GetFunctionDef (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_data_test.cc192 EXPECT_EQ(eager_context->GetFunctionDef("main"), nullptr); in TEST()
193 ASSERT_NE(eager_context->GetFunctionDef("add_subgraph"), nullptr); in TEST()
194 ASSERT_NE(eager_context->GetFunctionDef("mul_subgraph"), nullptr); in TEST()
196 *(eager_context->GetFunctionDef("add_subgraph")), add_fdef)); in TEST()
198 *(eager_context->GetFunctionDef("mul_subgraph")), mul_fdef)); in TEST()
233 EXPECT_EQ(eager_context->GetFunctionDef("main"), nullptr); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Doptimization_registry_test.cc73 const FunctionDef* GetFunctionDef(const string& func) const { in GetFunctionDef() function in tensorflow::OptimizationPassTest
83 auto f1 = GetFunctionDef("XTimesTwo"); in TEST_F()
87 auto f2 = GetFunctionDef("WXPlusB"); in TEST_F()
/external/tensorflow/tensorflow/c/eager/
Dgraph_function.cc23 Status GraphFunction::GetFunctionDef(FunctionDef** fdef) { in GetFunctionDef() function in tensorflow::tracing::graph::GraphFunction
Dgraph_function.h31 Status GetFunctionDef(FunctionDef** fdef) override;
Dabstract_function.h38 virtual Status GetFunctionDef(FunctionDef**) = 0;
/external/tensorflow/tensorflow/c/
Dc_api_function_test.cc231 EXPECT_TRUE(GetFunctionDef(func_, &fdef)); in fdef()
360 ASSERT_TRUE(GetFunctionDef(func_, &fdef)); in VerifyFDef()
371 ASSERT_TRUE(GetFunctionDef(func_, &fdef)); in Reincarnate()
953 ASSERT_TRUE(GetFunctionDef(func_, &fdef)); in TEST_F()
1547 ASSERT_TRUE(GetFunctionDef(func_, &fdef)); in TEST_F()
1555 ASSERT_TRUE(GetFunctionDef(func_, &fdef)); in TEST_F()
1563 ASSERT_TRUE(GetFunctionDef(func_, &fdef)); in TEST_F()
1640 ASSERT_TRUE(GetFunctionDef(f1, &fdef1)); in AssertEqual()
1641 ASSERT_TRUE(GetFunctionDef(f2, &fdef2)); in AssertEqual()
1649 GetFunctionDef(func, &fdef); in GetName()
Dc_test_util.h120 bool GetFunctionDef(TF_Function* func, tensorflow::FunctionDef* func_def);
Dc_test_util.cc412 bool GetFunctionDef(TF_Function* func, tensorflow::FunctionDef* func_def) { in GetFunctionDef() function
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcore.cc166 TF_RETURN_IF_ERROR(f->GetFunctionDef(&fdef)); in RegisterFunction()
Dcontext.h237 const FunctionDef* GetFunctionDef(const string& function_name);
Dcontext.cc798 const FunctionDef* EagerContext::GetFunctionDef(const string& function_name) { in GetFunctionDef() function in tensorflow::EagerContext
Dexecute.cc742 if (!op->EagerContext().GetFunctionDef(fname)) { in WrapInCallOp()
/external/tensorflow/tensorflow/cc/experimental/libtf/
Dfunction.cc52 TF_RETURN_IF_ERROR(trace->GetFunctionDef(&fdef)); in ExecuteFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc227 Status GetFunctionDef(tensorflow::FunctionDef** f) override;
519 Status MlirFunction::GetFunctionDef(tensorflow::FunctionDef** f) { in GetFunctionDef() function in mlir::TF::__anone669d3350111::MlirFunction
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.cc874 TF_RETURN_IF_ERROR(f->GetFunctionDef(&fdef)); in RegisterFunction()