| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_propagation.cc | 35 int64 memory_space = indexed_shape.shape.layout().memory_space(); in Run() local
45 int64 memory_space = indexed_shape.shape.layout().memory_space(); in Run() local
|
| D | memory_space_assignment.h | 47 AssignmentInformation* assignment_information_for_space(int64 memory_space) { in assignment_information_for_space()
524 Allocation(HloPosition defining_position, MemorySpace memory_space, in Allocation()
554 MemorySpace memory_space() const { return memory_space_; } in memory_space() function
1055 MemorySpaceAssignment::MemorySpace memory_space; member
|
| D | memory_space_assignment.cc | 1687 MemorySpaceAssignment::MemorySpace memory_space, int64 time, in AddRequiredAssignment()
1712 MemorySpace memory_space, AliasedOffset* offset) { in AddRequiredAssignment()
1734 MemorySpace memory_space = MemorySpace::kDefault; in AddInputAndOutputRequiredAssignments() local
1758 MemorySpace memory_space = MemorySpace::kDefault; in AddInputAndOutputRequiredAssignments() local
2113 MemorySpace memory_space, absl::optional<Chunk> chunk, int64 start_time, in AddAsyncCopy()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor_interface.h | 47 CreateTemporaryDeviceMemory(int64 memory_space, int64 byte_offset, in CreateTemporaryDeviceMemory()
|
| D | tpu_executor.h | 170 CreateTemporaryDeviceMemory(int64 memory_space, int64 byte_offset, in CreateTemporaryDeviceMemory()
|
| D | tpu_executor.cc | 195 DeviceMemoryBase TpuExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate()
|
| D | c_api_conversions.cc | 93 TF_Status* se_status) { in ToC()
|
| /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 | 811 int64 memory_space) { in AllocateArray()
|
| D | stream_executor_pimpl.cc | 487 DeviceMemoryBase StreamExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate()
899 int64 memory_space) { in Allocate()
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | op_metrics_db_combiner.cc | 90 uint64 memory_space = src_memory_accessed.memory_space(); in CombineMemoryAccessedBreakdown() local
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | op_metrics.proto | 64 uint64 memory_space = 2; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 37 int64 memory_space) { in Allocate()
|
| /external/llvm-project/polly/lib/External/ppcg/ |
| D | gpu_print.c | 83 const char *memory_space) in gpu_array_info_print_declaration_argument()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout.h | 213 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 | 178 int64 memory_space = 8; field
|
| D | shape_util.cc | 123 int64 element_size_in_bits, int64 memory_space) { in MakeShapeWithLayoutInternal()
312 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 | 69 DeviceMemoryBase HostExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.cc | 261 DeviceMemoryBase Allocate(uint64 size, int64 memory_space) override { in Allocate()
|
| D | stream_executor_test.cc | 50 int64_t memory_space, SP_DeviceMemoryBase* const mem) {} in allocate()
346 int64_t memory_space, SP_DeviceMemoryBase* const mem) { in TEST_F()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 517 DeviceMemoryBase GpuExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 425 DeviceMemoryBase GpuExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate()
|