| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_platform_info.cc | 112 absl::optional<se::TfAllocatorAdapter>* tf_allocator_adapter, in GetAllocator() argument
123 tf_allocator_adapter->emplace(device->GetAllocator({}), platform); in GetAllocator()
124 return &tf_allocator_adapter->value(); in GetAllocator()
126 tf_allocator_adapter->emplace(device->GetAllocator({}), stream); in GetAllocator()
127 return &tf_allocator_adapter->value(); in GetAllocator()
134 absl::optional<se::TfAllocatorAdapter>* tf_allocator_adapter) { in GenerateCompilerOptions() argument
146 GetAllocator(tf_allocator_adapter, device, stream, platform_info); in GenerateCompilerOptions()
|
| D | xla_compile_on_demand_op.cc | 51 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in Run() local
53 &tf_allocator_adapter, ctx->device(), in Run()
129 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in Compile() local
134 /*has_ref_vars=*/true, &tf_allocator_adapter); in Compile()
|
| D | xla_platform_info.h | 98 absl::optional<se::TfAllocatorAdapter>* tf_allocator_adapter,
108 absl::optional<se::TfAllocatorAdapter>* tf_allocator_adapter);
|
| D | get_compiler_ir.cc | 103 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in GetCompilerIr() local
108 /*has_ref_vars=*/false, &tf_allocator_adapter); in GetCompilerIr()
|
| D | BUILD | 273 "//tensorflow/stream_executor:tf_allocator_adapter",
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 193 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in CompileToLocalExecutable() local
197 platform_info, has_ref_vars, &tf_allocator_adapter); in CompileToLocalExecutable()
250 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in Compute() local
252 &tf_allocator_adapter, ctx->device(), in Compute()
466 absl::optional<se::TfAllocatorAdapter> tf_allocator_adapter; in Compute() local
468 &tf_allocator_adapter, ctx->device(), in Compute()
|
| D | BUILD | 39 "//tensorflow/stream_executor:tf_allocator_adapter",
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | BUILD | 682 name = "tf_allocator_adapter",
683 srcs = ["tf_allocator_adapter.cc"],
684 hdrs = ["tf_allocator_adapter.h"],
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 393 se::TfAllocatorAdapter tf_allocator_adapter(ctx->device()->GetAllocator({}), in operator ()() local
396 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()()
413 stream, &tf_allocator_adapter, se::GpuAsmOpts(), in operator ()()
|
| D | conv_ops_3d.cc | 510 se::TfAllocatorAdapter tf_allocator_adapter( in launch() local
512 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in launch()
533 stream, &tf_allocator_adapter, se::GpuAsmOpts(), in launch()
|
| D | conv_ops_fused_impl.h | 386 se::TfAllocatorAdapter tf_allocator_adapter(
388 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter,
397 stream, &tf_allocator_adapter, se::GpuAsmOpts(),
|
| D | conv_grad_filter_ops.cc | 988 se::TfAllocatorAdapter tf_allocator_adapter(ctx->device()->GetAllocator({}), in operator ()() local
990 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()()
1007 stream, &tf_allocator_adapter, se::GpuAsmOpts(), in operator ()()
|
| D | conv_ops.cc | 1006 se::TfAllocatorAdapter tf_allocator_adapter(ctx->device()->GetAllocator({}), in operator ()() local
1008 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()()
1018 stream, &tf_allocator_adapter, se::GpuAsmOpts(), in operator ()()
|
| D | conv_grad_ops_3d.cc | 1428 se::TfAllocatorAdapter tf_allocator_adapter( in Compute() local
1430 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in Compute()
1447 stream, &tf_allocator_adapter, se::GpuAsmOpts(), in Compute()
|
| D | BUILD | 3802 "//tensorflow/stream_executor:tf_allocator_adapter",
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | BUILD | 314 "//tensorflow/stream_executor:tf_allocator_adapter",
|