Home
last modified time | relevance | path

Searched refs:RemoveFunction (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/gradients/tape/
Dtape_context.cc42 Status TapeContext::RemoveFunction(const string& func) { in RemoveFunction() function in tensorflow::gradients::TapeContext
43 return parent_ctx_->RemoveFunction(func); in RemoveFunction()
Dtape_context.h29 Status RemoveFunction(const string& func) override;
/external/tensorflow/tensorflow/c/experimental/op_handler/
Dinternal.cc38 Status OpHandlerContext::RemoveFunction(const string& function) { in RemoveFunction() function in tensorflow::OpHandlerContext
39 return parent_ctx_->RemoveFunction(function); in RemoveFunction()
Dinternal.h93 Status RemoveFunction(const string&) override;
/external/tensorflow/tensorflow/c/eager/
Dabstract_context.h61 virtual Status RemoveFunction(const string& func) = 0;
Dunified_api_testutil.cc115 TF_RETURN_IF_ERROR(ctx->RemoveFunction(fn_name)); in RunModel()
Dgradients_util.cc301 TF_RETURN_IF_ERROR(ctx->RemoveFunction(fn_name)); in RunModel()
Dc_api_unified_experimental_graph.cc419 Status RemoveFunction(const string& func) override { in RemoveFunction() function in tensorflow::tracing::graph::GraphContext
Dc_api.cc946 status->status = tensorflow::unwrap(ctx)->RemoveFunction(name); in TFE_ContextRemoveFunction()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dflat_tensor_function.cc41 Status status = ctx_->RemoveFunction(name_); in ~FlatTensorFunction()
/external/tensorflow/tensorflow/python/framework/experimental/
Ddef_function.py68 outer_ctx.RemoveFunction(self.name)
Dunified_api.cc128 Status s = self->RemoveFunction(func); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/core/framework/
Dfunction_test.cc1057 TEST(FunctionLibraryDefinitionTest, RemoveFunction) { in TEST() argument
1061 Status s = lib_def.RemoveFunction("XTimes16"); in TEST()
1067 TF_EXPECT_OK(lib_def.RemoveFunction("XTimesTwo")); in TEST()
Dfunction.h438 Status RemoveFunction(const std::string& func) TF_LOCKS_EXCLUDED(mu_);
Dfunction.cc1444 Status FunctionLibraryDefinition::RemoveFunction(const string& func) { in RemoveFunction() function in tensorflow::FunctionLibraryDefinition
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc445 TF_RETURN_IF_ERROR(base->RemoveFunction(fn)); in AddToFunctionLibrary()
461 TF_RETURN_IF_ERROR(base->RemoveFunction(fd.signature().name())); in AddToFunctionLibrary()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h234 Status RemoveFunction(const string& func) override;
Dcontext.cc767 Status EagerContext::RemoveFunction(const string& func) { in RemoveFunction() function in tensorflow::EagerContext
787 return func_lib_def_.RemoveFunction(func); in RemoveFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc272 Status RemoveFunction(const string& func) override { in RemoveFunction() function in mlir::TF::__anon3f449e090111::MlirFunctionContext
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc788 TF_RETURN_IF_ERROR(fld->RemoveFunction(func_name)); in PruneUnreachableFunctionsFromGraph()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc2469 TF_RETURN_IF_ERROR(fld->RemoveFunction(func)); in ExtractOutsideCompilationForFunction()
2534 TF_RETURN_IF_ERROR(fld->RemoveFunction(host_graph_func_name)); in ExtractOutsideCompilation()