/external/tensorflow/tensorflow/compiler/xla/service/ |
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
|
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
|
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()
|
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
D | tpu_executor_interface.h | 47 CreateTemporaryDeviceMemory(int64_t memory_space, int64_t byte_offset, in CreateTemporaryDeviceMemory()
|
D | tpu_executor.h | 175 CreateTemporaryDeviceMemory(int64_t memory_space, int64_t byte_offset, in CreateTemporaryDeviceMemory()
|
D | tpu_executor.cc | 195 DeviceMemoryBase TpuExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
|
D | c_api_conversions.cc | 99 TF_Status* se_status) { in ToC()
|
/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
D | tf_allocator_adapter.h | 95 int64_t memory_space) override { in Allocate()
|
D | tf_allocator_adapter.cc | 40 int64_t memory_space) { in Allocate()
|
D | stream_executor_pimpl.h | 829 int64_t memory_space) { in AllocateArray()
|
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()
|
/external/tensorflow/tensorflow/core/profiler/convert/ |
D | op_metrics_db_combiner.cc | 98 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_t memory_space) { in Allocate()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | layout.h | 228 int64_t memory_space() const { return memory_space_; } in memory_space() function
|
D | layout_util.cc | 60 int64_t memory_space) { in MakeLayout()
|
D | xla_data.proto | 174 int64 memory_space = 8; field
|
D | shape_util.cc | 104 int64_t memory_space) { in MakeShapeWithLayoutInternal() 317 int64_t memory_space) { in MakeShapeWithLayout()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | local_client_test_base.cc | 45 int64_t memory_space) { in Allocate()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | while_op_test.cc | 134 int64_t memory_space, SP_DeviceMemoryBase* const mem) { in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
D | host_gpu_executor.cc | 69 DeviceMemoryBase HostExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
|
/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
D | stream_executor_test_util.cc | 24 int64_t memory_space, SP_DeviceMemoryBase* const mem) {} in Allocate()
|
D | stream_executor.cc | 233 DeviceMemoryBase Allocate(uint64 size, int64_t memory_space) override { in Allocate()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_gpu_executor.cc | 427 DeviceMemoryBase GpuExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
|
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
D | cuda_gpu_executor.cc | 559 DeviceMemoryBase GpuExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
|