Searched refs:TFConcreteFunction (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
D | tf_concrete_function.h | 38 class TFConcreteFunction : public ConcreteFunction { 58 std::unique_ptr<TFConcreteFunction>* out); 66 ~TFConcreteFunction() override = default; 69 TFConcreteFunction(std::unique_ptr<FlatTensorFunction> func, 72 TFConcreteFunction(const TFConcreteFunction&) = delete; 73 TFConcreteFunction& operator=(const TFConcreteFunction&) = delete;
|
D | tf_concrete_function.cc | 36 TFConcreteFunction::TFConcreteFunction(std::unique_ptr<FlatTensorFunction> func, in TFConcreteFunction() function in tensorflow::TFConcreteFunction 40 Status TFConcreteFunction::Create( in Create() 44 std::unique_ptr<TFConcreteFunction>* out) { in Create() 49 out->reset(new TFConcreteFunction(std::move(func), std::move(metadata))); in Create() 53 const FunctionMetadata& TFConcreteFunction::GetFunctionMetadata() const { in GetFunctionMetadata() 57 Status TFConcreteFunction::MakeCallOp( in MakeCallOp()
|
D | restored_resource.h | 65 TFConcreteFunction* create_resource, 66 TFConcreteFunction* initialize, 67 TFConcreteFunction* destroy_resource, 80 TFConcreteFunction* create_resource_; 81 TFConcreteFunction* initialize_; 82 TFConcreteFunction* destroy_resource_;
|
D | restored_resource.cc | 32 Status ExecuteNoArgDummyReturnFunction(TFConcreteFunction* func) { in ExecuteNoArgDummyReturnFunction() 47 TFConcreteFunction* create_resource, in RestoredResource() 48 TFConcreteFunction* initialize, in RestoredResource() 49 TFConcreteFunction* destroy_resource, in RestoredResource()
|
D | partially_revived_objects.cc | 283 std::unique_ptr<TFConcreteFunction>* out) { in CreateConcreteFunction() 294 return TFConcreteFunction::Create(/*function_def=*/builder.fdef, in CreateConcreteFunction() 343 std::unique_ptr<TFConcreteFunction> out; in InitializeCreateResourceFunctions() 370 std::unique_ptr<TFConcreteFunction> out; in InitializeAllFunctions() 401 const TFConcreteFunction* create_resource_fn = in CreateAllResourceHandles() 438 TFConcreteFunction* create_resource = nullptr; in BuildResources() 452 TFConcreteFunction* initialize = nullptr; in BuildResources() 463 TFConcreteFunction* destroy_resource = nullptr; in BuildResources()
|
D | revived_objects.h | 85 RevivedObjectContainer<TFConcreteFunction> concrete_functions;
|
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
D | tf_concrete_function_loading_test.cc | 81 std::unique_ptr<TFConcreteFunction> result; in TEST_F() 105 std::unique_ptr<TFConcreteFunction> result; in TEST_F() 124 std::unique_ptr<TFConcreteFunction> result; in TEST_F() 149 std::unique_ptr<TFConcreteFunction> result; in TEST_F() 174 std::unique_ptr<TFConcreteFunction> result; in TEST_F() 193 std::unique_ptr<TFConcreteFunction> result; in TEST_F() 213 std::unique_ptr<TFConcreteFunction> result; in TEST_F() 238 std::unique_ptr<TFConcreteFunction> result; in TEST_F() 257 std::unique_ptr<TFConcreteFunction> result; in TEST_F()
|
D | tf_saved_model_api.cc | 274 std::unique_ptr<TFConcreteFunction> concrete_function; in Load() 275 TF_RETURN_IF_ERROR(TFConcreteFunction::Create(/*function_def=*/&function, in Load()
|
D | saved_model_utils.h | 72 ImmediateExecutionContext* ctx, std::unique_ptr<TFConcreteFunction>* out);
|
D | saved_model_utils.cc | 257 ImmediateExecutionContext* ctx, std::unique_ptr<TFConcreteFunction>* out) { in LoadTFConcreteFunction() 274 return TFConcreteFunction::Create(function_def, std::move(captures), {}, ctx, in LoadTFConcreteFunction()
|