Home
last modified time | relevance | path

Searched defs:memory_space (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_propagation.cc43 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
Dmemory_space_assignment.h67 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
Dmemory_space_assignment.cc2361 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/
Dtpu_executor_interface.h47 CreateTemporaryDeviceMemory(int64_t memory_space, int64_t byte_offset, in CreateTemporaryDeviceMemory()
Dtpu_executor.h175 CreateTemporaryDeviceMemory(int64_t memory_space, int64_t byte_offset, in CreateTemporaryDeviceMemory()
Dtpu_executor.cc195 DeviceMemoryBase TpuExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
Dc_api_conversions.cc99 TF_Status* se_status) { in ToC()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dtf_allocator_adapter.h95 int64_t memory_space) override { in Allocate()
Dtf_allocator_adapter.cc40 int64_t memory_space) { in Allocate()
Dstream_executor_pimpl.h829 int64_t memory_space) { in AllocateArray()
Dstream_executor_pimpl.cc497 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/
Dop_metrics_db_combiner.cc98 uint64 memory_space = src_memory_accessed.memory_space(); in CombineMemoryAccessedBreakdown() local
/external/tensorflow/tensorflow/core/profiler/protobuf/
Dop_metrics.proto64 uint64 memory_space = 2; field
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.cc37 int64_t memory_space) { in Allocate()
/external/tensorflow/tensorflow/compiler/xla/
Dlayout.h228 int64_t memory_space() const { return memory_space_; } in memory_space() function
Dlayout_util.cc60 int64_t memory_space) { in MakeLayout()
Dxla_data.proto174 int64 memory_space = 8; field
Dshape_util.cc104 int64_t memory_space) { in MakeShapeWithLayoutInternal()
317 int64_t memory_space) { in MakeShapeWithLayout()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc45 int64_t memory_space) { in Allocate()
/external/tensorflow/tensorflow/core/kernels/
Dwhile_op_test.cc134 int64_t memory_space, SP_DeviceMemoryBase* const mem) { in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
Dhost_gpu_executor.cc69 DeviceMemoryBase HostExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test_util.cc24 int64_t memory_space, SP_DeviceMemoryBase* const mem) {} in Allocate()
Dstream_executor.cc233 DeviceMemoryBase Allocate(uint64 size, int64_t memory_space) override { in Allocate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc427 DeviceMemoryBase GpuExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_gpu_executor.cc559 DeviceMemoryBase GpuExecutor::Allocate(uint64_t size, int64_t memory_space) { in Allocate()

12