Home
last modified time | relevance | path

Searched refs:StubResource (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dresource_op_kernel_test.cc47 class StubResource : public ResourceBase { class
53 class StubResourceOpKernel : public ResourceOpKernel<StubResource> {
57 StubResource* resource() TF_LOCKS_EXCLUDED(mu_) { in resource()
63 Status CreateResource(StubResource** resource) override { in CreateResource()
64 *resource = CHECK_NOTNULL(new StubResource); in CreateResource()
68 Status VerifyResource(StubResource* resource) override { in VerifyResource()
145 StubResource* resource; in TEST_F()
147 mgr_.Lookup<StubResource>(mgr_.default_container(), key, &resource)); in TEST_F()
155 mgr_.Lookup<StubResource>(mgr_.default_container(), key, &resource); in TEST_F()
167 StubResource* resource; in TEST_F()
[all …]
Dresource_mgr_test.cc253 class StubResource : public ResourceBase { class
268 MakeResourceHandle<StubResource>(&ctx, "container", "name"); in TEST()
271 auto* r = new StubResource(); in TEST()
276 core::RefCountPtr<StubResource> r; in TEST()
282 TF_EXPECT_OK(DeleteResource<StubResource>(&ctx, p)); in TEST()
283 core::RefCountPtr<StubResource> unused; in TEST()
297 MakeResourceHandle<StubResource>(&ctx, "container", "name"); in TEST()
306 auto* r = new StubResource(); in TEST()
326 MakeResourceHandle<StubResource>(&ctx, "container", "name"); in TEST()
342 MakeResourceHandle<StubResource>(&ctx, "container", "name"); in TEST()
[all …]
/external/tensorflow/tensorflow/python/framework/
Dtest_ops.cc58 REGISTER_RESOURCE_HANDLE_OP(StubResource);
164 class StubResource : public ResourceBase { class
169 REGISTER_RESOURCE_HANDLE_KERNEL(StubResource);
172 IsResourceInitialized<StubResource>);
180 CreateResource(c, HandleFromInput(c, 0), new StubResource)); in Compute()
193 StubResource* unused; in Compute()
194 OP_REQUIRES_OK(ctx, LookupResource<StubResource>( in Compute()