| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_handle.cc | 22 ResourceHandle::ResourceHandle() {} in ResourceHandle() function in tensorflow::ResourceHandle
24 ResourceHandle::ResourceHandle(const ResourceHandleProto& proto) { in ResourceHandle() function in tensorflow::ResourceHandle
28 ResourceHandle::~ResourceHandle() {} in ~ResourceHandle()
30 void ResourceHandle::AsProto(ResourceHandleProto* proto) const { in AsProto()
38 void ResourceHandle::FromProto(const ResourceHandleProto& proto) { in FromProto()
46 string ResourceHandle::SerializeAsString() const { in SerializeAsString()
52 bool ResourceHandle::ParseFromString(const string& s) { in ParseFromString()
59 string ResourceHandle::DebugString() const { in DebugString()
65 string ProtoDebugString(const ResourceHandle& handle) { in ProtoDebugString()
69 void EncodeResourceHandleList(const ResourceHandle* p, int64 n, in EncodeResourceHandleList()
[all …]
|
| D | resource_handle.h | 32 class ResourceHandle {
34 ResourceHandle();
35 ResourceHandle(const ResourceHandleProto& proto);
36 ~ResourceHandle();
84 string ProtoDebugString(const ResourceHandle& handle);
87 void EncodeResourceHandleList(const ResourceHandle* p, int64 n,
92 ResourceHandle* ps, int64 n);
|
| D | resource_mgr.h | 170 Status Delete(const ResourceHandle& handle) TF_MUST_USE_RESULT;
236 ResourceHandle MakeResourceHandle(OpKernelContext* ctx, const string& container,
241 ResourceHandle MakeResourceHandle(OpKernelContext* ctx, const string& container, in MakeResourceHandle()
251 ResourceHandle MakePerStepResourceHandle(OpKernelContext* ctx,
255 const ResourceHandle& HandleFromInput(OpKernelContext* ctx, int input);
257 ResourceHandle* handle);
264 Status CreateResource(OpKernelContext* ctx, const ResourceHandle& p, T* value);
271 Status LookupResource(OpKernelContext* ctx, const ResourceHandle& p, T** value);
277 OpKernelContext* ctx, absl::Span<ResourceHandle const> p,
287 Status LookupOrCreateResource(OpKernelContext* ctx, const ResourceHandle& p,
[all …]
|
| D | resource_mgr.cc | 35 ResourceHandle MakeResourceHandle(OpKernelContext* ctx, const string& container, in MakeResourceHandle()
38 ResourceHandle result; in MakeResourceHandle()
47 if (name == ResourceHandle::ANONYMOUS_NAME) { in MakeResourceHandle()
63 handle->scalar<ResourceHandle>()() = in MakeResourceHandleToOutput()
70 Status ValidateDevice(OpKernelContext* ctx, const ResourceHandle& p) { in ValidateDevice()
211 Status ResourceMgr::Delete(const ResourceHandle& handle) { in Delete()
291 const ResourceHandle& HandleFromInput(OpKernelContext* ctx, int input) { in HandleFromInput()
292 return ctx->input(input).flat<ResourceHandle>()(0); in HandleFromInput()
296 ResourceHandle* handle) { in HandleFromInput()
299 *handle = tensor->flat<ResourceHandle>()(0); in HandleFromInput()
[all …]
|
| D | allocator.h | 235 virtual void RunResourceCtor(ResourceHandle* p, size_t n) { in RunResourceCtor()
236 for (size_t i = 0; i < n; ++p, ++i) new (p) ResourceHandle(); in RunResourceCtor()
240 virtual void RunResourceDtor(ResourceHandle* p, size_t n) { in RunResourceDtor()
241 for (size_t i = 0; i < n; ++p, ++i) p->~ResourceHandle(); in RunResourceDtor()
266 inline void Allocator::RunCtor(ResourceHandle* p, size_t n) { in RunCtor()
271 inline void Allocator::RunDtor(ResourceHandle* p, size_t n) { in RunDtor()
|
| D | allocator.cc | 44 void RunResourceCtor(ResourceHandle* p, size_t n) { in RunResourceCtor()
45 for (size_t i = 0; i < n; ++p, ++i) new (p) ResourceHandle(); in RunResourceCtor()
48 void RunResourceDtor(ResourceHandle* p, size_t n) { in RunResourceDtor()
49 for (size_t i = 0; i < n; ++p, ++i) p->~ResourceHandle(); in RunResourceDtor()
|
| D | resource_mgr_test.cc | 260 ResourceHandle p = in TEST()
290 ResourceHandle p = in TEST()
319 ResourceHandle p = in TEST()
335 ResourceHandle p = in TEST()
|
| D | tensor.cc | 206 struct Helper<ResourceHandle> {
214 EncodeResourceHandleList(in->base<const ResourceHandle>(), n, in Encode()
223 auto* buf = new Buffer<ResourceHandle>(a, n); in Decode()
224 ResourceHandle* ps = buf->template base<ResourceHandle>(); in Decode()
236 return n * sizeof(ResourceHandle); in TotalBytes()
339 struct ProtoHelper<ResourceHandle> {
347 static void Fill(const ResourceHandle* data, size_t n, TensorProto* proto) { in Fill()
719 CASE(ResourceHandle, SINGLE_ARG(STMTS)) \
|
| D | resource_handle.proto | 5 option java_outer_classname = "ResourceHandle";
|
| /external/tensorflow/tensorflow/core/kernels/lookup_tables/ |
| D | op_kernel_templates.h | 99 handle->scalar<ResourceHandle>()() = MakeResourceHandle<ResourceType>( in Compute()
186 ResourceHandle handle(table_handle.scalar<ResourceHandle>()()); in Compute()
234 handle->scalar<ResourceHandle>()() = MakeResourceHandle<ResourceType>( in Compute()
312 ResourceHandle handle(table_handle.scalar<ResourceHandle>()()); in Compute()
393 ResourceHandle handle(table_handle.scalar<ResourceHandle>()()); in Compute()
426 ResourceHandle handle(container_handle.scalar<ResourceHandle>()()); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_cpu.cc | 39 struct MemCpyCopier<ResourceHandle> {
40 inline void Copy(ResourceHandle* dst, const ResourceHandle* src, in Copy()
|
| D | control_flow_ops.cc | 117 REGISTER_GPU_HOST_KERNEL(ResourceHandle);
319 REGISTER_GPU_HOST_KERNEL(ResourceHandle);
342 REGISTER_SYCL_HOST_KERNEL(ResourceHandle);
409 REGISTER_SYCL_HOST_KERNEL(ResourceHandle);
438 REGISTER_GPU_HOST_KERNEL(ResourceHandle);
520 REGISTER_GPU_HOST_KERNEL(ResourceHandle);
570 REGISTER_GPU_HOST_KERNEL(ResourceHandle);
|
| D | queue_ops.cc | 68 const ResourceHandle& ref = context->input(0).flat<ResourceHandle>()(0); in Compute()
|
| D | strided_slice_op.h | 63 struct InitOutput<ResourceHandle, NDIMS, Device> {
65 typename TTypes<ResourceHandle, NDIMS>::Tensor output) {
66 output.device(d) = output.constant(ResourceHandle());
|
| D | session_ops.cc | 52 ResourceHandle resource_handle = MakeResourceHandle<Tensor>( in Compute()
57 handle->scalar<ResourceHandle>()() = resource_handle; in Compute()
|
| D | ops_testutil.h | 147 ResourceHandle handle; in AddResourceInput()
154 input->scalar<ResourceHandle>()() = handle; in AddResourceInput()
|
| D | partitioned_function_ops.cc | 193 const ResourceHandle& handle = tensor.flat<ResourceHandle>()(0); in Instantiate()
|
| D | identity_op.cc | 148 REGISTER_GPU_HOST_KERNEL(ResourceHandle);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 71 void RunResourceCtor(ResourceHandle* p, size_t n) override {} in RunResourceCtor()
72 void RunResourceDtor(ResourceHandle* p, size_t n) override {} in RunResourceDtor()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.h | 109 .TypeConstraint<ResourceHandle>("T") \
215 .TypeConstraint<ResourceHandle>("T"), \
224 .TypeConstraint<ResourceHandle>("T") \
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_GetSessionHandleV2.pbtxt | 13 as a ResourceHandle object.
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 417 if (name_ == ResourceHandle::ANONYMOUS_NAME) { in Compute()
595 resource_handle_t.scalar<ResourceHandle>()().SerializeAsString(); in Compute()
622 ResourceHandle resource_handle; in Compute()
653 resource_handle_t->scalar<ResourceHandle>()() = resource_handle; in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | quantile_ops.cc | 224 ResourceHandle handle; in Compute()
336 ResourceHandle handle; in Compute()
391 ResourceHandle handle; in Compute()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 322 const ResourceHandle& resource_handle = t.flat<ResourceHandle>()(0); in GetResourceDevice()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | stats_accumulator_ops.cc | 451 const ResourceHandle& handle = in Compute()
453 .flat<ResourceHandle>()(0); in Compute()
516 const ResourceHandle& handle = in Compute()
518 .flat<ResourceHandle>()(0); in Compute()
|