Searched refs:TensorFlowWrapper (Results 1 – 5 of 5) sorted by relevance
39 TensorFlowWrapper::InitializeConfigProto(const Any& external_config_proto) { in InitializeConfigProto()83 absl::StatusOr<std::unique_ptr<TensorFlowWrapper>> TensorFlowWrapper::Create( in Create()127 auto wrapper = absl::WrapUnique(new TensorFlowWrapper( in Create()132 TensorFlowWrapper::~TensorFlowWrapper() { FCP_CHECK(CloseAndRelease().ok()); } in ~TensorFlowWrapper()134 absl::Status TensorFlowWrapper::ToFcpStatus(tensorflow::Status s, in ToFcpStatus()146 absl::Status TensorFlowWrapper::Run( in Run()173 absl::Status TensorFlowWrapper::CloseAndRelease() { in CloseAndRelease()
41 TensorFlowWrapper::InitializeConfigProto(packed_config_proto); in TEST()55 TensorFlowWrapper::InitializeConfigProto(packed_config_proto); in TEST()67 TensorFlowWrapper::InitializeConfigProto(packed_config_proto); in TEST()80 TensorFlowWrapper::InitializeConfigProto(packed_config_proto); in TEST()93 TensorFlowWrapper::InitializeConfigProto(packed_config_proto); in TEST()113 TensorFlowWrapper::InitializeConfigProto(packed_config_proto); in TEST()
47 class TensorFlowWrapper {49 static absl::StatusOr<std::unique_ptr<TensorFlowWrapper>> Create(61 ~TensorFlowWrapper();78 TensorFlowWrapper(std::unique_ptr<tensorflow::Session> session, in TensorFlowWrapper() function
74 absl::StatusOr<std::unique_ptr<TensorFlowWrapper>> tf_wrapper_or = in RunPlan()75 TensorFlowWrapper::Create(graph, config_proto, should_abort_, in RunPlan()81 std::unique_ptr<TensorFlowWrapper> tf_wrapper = in RunPlan()116 TensorFlowWrapper* tf_wrapper, in RunPlanInternal()162 TensorFlowWrapper* tf_wrapper, in RunTensorFlowInternal()
72 TensorFlowWrapper* tf_wrapper,89 TensorFlowWrapper* tf_wrapper,