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()
|