Searched defs:memory_space (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment.h | 45 AssignmentInformation* assignment_information_for_space(int64 memory_space) { in assignment_information_for_space()
357 MemorySpace memory_space, Chunk chunk, int64 start_time, in Allocation()
393 MemorySpace memory_space() const { return memory_space_; } in memory_space() function
418 CopyAllocation(const Allocation& prev_allocation, MemorySpace memory_space, in CopyAllocation()
573 MemorySpaceAssignment::MemorySpace memory_space; member
|
| D | memory_space_assignment.cc | 538 MemorySpace memory_space = MemorySpace::kDefault; in AddInputAndOutputRequiredAssignments() local
562 MemorySpace memory_space = MemorySpace::kDefault; in AddInputAndOutputRequiredAssignments() local
894 MemorySpace memory_space, Chunk chunk, int64 start_time, int64 end_time, in AddAsyncCopy()
|
| D | layout_assignment.cc | 1978 int64 memory_space = defining_shape.layout().memory_space(); in PropagateMemorySpace() local
|
| D | hlo_parser.cc | 3614 tensorflow::int64 memory_space = 0; in ParseLayout() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | tf_allocator_adapter.h | 83 int64 memory_space) override { in Allocate()
|
| D | tf_allocator_adapter.cc | 40 int64 memory_space) { in Allocate()
|
| D | stream_executor_pimpl.h | 800 int64 memory_space) { in AllocateArray()
|
| D | stream_executor_pimpl.cc | 476 DeviceMemoryBase StreamExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate()
888 int64 memory_space) { in Allocate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 37 int64 memory_space) { in Allocate()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout.h | 212 int64 memory_space() const { return memory_space_; } in memory_space() function
|
| D | layout_util.cc | 60 int64 element_size_in_bits, int64 memory_space) { in MakeLayout()
|
| D | xla_data.proto | 177 int64 memory_space = 8; field
|
| D | shape_util.cc | 94 int64 element_size_in_bits, int64 memory_space) { in MakeShapeWithLayoutInternal()
219 int64 element_size_in_bits, int64 memory_space) { in MakeShapeWithLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 47 int64 memory_space) { in Allocate()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 45 DeviceMemoryBase HostExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 517 DeviceMemoryBase GpuExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate()
|
| D | cuda_driver.cc | 170 string MemorySpaceString(MemorySpace memory_space) { in MemorySpaceString()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 420 DeviceMemoryBase GpuExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate()
|
| D | rocm_driver.cc | 133 string MemorySpaceString(MemorySpace memory_space) { in MemorySpaceString()
|