| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | memory_space_propagation.cc | 43 int64_t memory_space = indexed_shape.shape.layout().memory_space(); in Run() local
53 int64_t memory_space = indexed_shape.shape.layout().memory_space(); in Run() local
|
| H A D | memory_space_assignment.h | 67 int64_t memory_space) { in assignment_information_for_space()
541 Allocation(HloPosition defining_position, MemorySpace memory_space, in Allocation()
590 MemorySpace memory_space() const { return memory_space_; } in memory_space() function
1317 MemorySpaceAssignment::MemorySpace memory_space; member
|
| H A D | memory_space_assignment.cc | 2361 MemorySpaceAssignment::MemorySpace memory_space, int64_t time, in AddRequiredAssignment()
2386 MemorySpace memory_space, AliasedOffset* offset) { in AddRequiredAssignment()
2408 MemorySpace memory_space = MemorySpace::kDefault; in AddInputAndOutputRequiredAssignments() local
2432 MemorySpace memory_space = MemorySpace::kDefault; in AddInputAndOutputRequiredAssignments() local
2889 MemorySpace memory_space, std::optional<Chunk> chunk, int64_t start_time, in AddAsyncCopy()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | tpu_executor_interface.h | 47 CreateTemporaryDeviceMemory(int64_t memory_space, int64_t byte_offset, in CreateTemporaryDeviceMemory()
|
| H A D | tpu_executor.h | 175 CreateTemporaryDeviceMemory(int64_t memory_space, int64_t byte_offset, in CreateTemporaryDeviceMemory()
|
| H A D | tpu_executor.cc | 195 DeviceMemoryBase TpuExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
|
| H A D | c_api_conversions.cc | 99 TF_Status* se_status) { in ToC()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | tf_allocator_adapter.h | 95 int64_t memory_space) override { in Allocate()
|
| H A D | tf_allocator_adapter.cc | 40 int64_t memory_space) { in Allocate()
|
| H A D | stream_executor_pimpl.h | 829 int64_t memory_space) { in AllocateArray()
|
| H A D | stream_executor_pimpl.cc | 497 DeviceMemoryBase StreamExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
913 int64_t memory_space) { in Allocate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/ |
| H A D | op_metrics_db_combiner.cc | 98 uint64 memory_space = src_memory_accessed.memory_space(); in CombineMemoryAccessedBreakdown() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/protobuf/ |
| H A D | op_metrics.proto | 64 uint64 memory_space = 2; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | executor.cc | 37 int64_t memory_space) { in Allocate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | layout.h | 228 int64_t memory_space() const { return memory_space_; } in memory_space() function
|
| H A D | layout_util.cc | 60 int64_t memory_space) { in MakeLayout()
|
| H A D | shape_util.cc | 104 int64_t memory_space) { in MakeShapeWithLayoutInternal()
317 int64_t memory_space) { in MakeShapeWithLayout()
|
| H A D | xla_data.proto | 174 int64 memory_space = 8; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | local_client_test_base.cc | 45 int64_t memory_space) { in Allocate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | while_op_test.cc | 134 int64_t memory_space, SP_DeviceMemoryBase* const mem) { in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| H A D | host_gpu_executor.cc | 69 DeviceMemoryBase HostExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| H A D | stream_executor_test_util.cc | 24 int64_t memory_space, SP_DeviceMemoryBase* const mem) {} in Allocate()
|
| H A D | stream_executor.cc | 233 DeviceMemoryBase Allocate(uint64 size, int64_t memory_space) override { in Allocate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_gpu_executor.cc | 427 DeviceMemoryBase GpuExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_gpu_executor.cc | 559 DeviceMemoryBase GpuExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
|