Searched defs:tf_allocator_adapter (Results 1 – 9 of 9) sorted by relevance
112 absl::optional<se::TfAllocatorAdapter>* tf_allocator_adapter, in GetAllocator()134 absl::optional<se::TfAllocatorAdapter>* tf_allocator_adapter) { in GenerateCompilerOptions()
51 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in Run() local129 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in Compile() local
103 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in GetCompilerIr() local
193 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in CompileToLocalExecutable() local250 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in Compute() local466 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in Compute() local
393 se::TfAllocatorAdapter tf_allocator_adapter(ctx->device()->GetAllocator({}), in operator ()() local
510 se::TfAllocatorAdapter tf_allocator_adapter( in launch() local
988 se::TfAllocatorAdapter tf_allocator_adapter(ctx->device()->GetAllocator({}), in operator ()() local
1006 se::TfAllocatorAdapter tf_allocator_adapter(ctx->device()->GetAllocator({}), in operator ()() local
1428 se::TfAllocatorAdapter tf_allocator_adapter( in Compute() local