| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_propagation.cc | 40 int64_t memory_space = indexed_shape.shape.layout().memory_space(); in Run() local
50 int64_t memory_space = indexed_shape.shape.layout().memory_space(); in Run() local
|
| D | memory_space_assignment.h | 61 int64_t memory_space) { in assignment_information_for_space()
480 Allocation(HloPosition defining_position, MemorySpace memory_space, in Allocation()
529 MemorySpace memory_space() const { return memory_space_; } in memory_space() function
1196 MemorySpaceAssignment::MemorySpace memory_space; member
|
| D | memory_space_assignment.cc | 1884 MemorySpaceAssignment::MemorySpace memory_space, int64_t time, in AddRequiredAssignment()
1909 MemorySpace memory_space, AliasedOffset* offset) { in AddRequiredAssignment()
1931 MemorySpace memory_space = MemorySpace::kDefault; in AddInputAndOutputRequiredAssignments() local
1955 MemorySpace memory_space = MemorySpace::kDefault; in AddInputAndOutputRequiredAssignments() local
2344 MemorySpace memory_space, absl::optional<Chunk> chunk, int64_t start_time, in AddAsyncCopy()
|
| /external/tensorflow/tensorflow/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 | c_api_conversions.cc | 93 TF_Status* se_status) { in ToC()
|
| D | tpu_executor.cc | 195 DeviceMemoryBase TpuExecutor::Allocate(uint64 size, int64_t memory_space) { in Allocate()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | tf_allocator_adapter.h | 83 int64_t memory_space) override { in Allocate()
|
| D | tf_allocator_adapter.cc | 40 int64_t memory_space) { in Allocate()
|
| D | stream_executor_pimpl.h | 850 int64_t memory_space) { in AllocateArray()
|
| D | stream_executor_pimpl.cc | 494 DeviceMemoryBase StreamExecutor::Allocate(uint64 size, int64_t memory_space) { in Allocate()
910 int64_t 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_t memory_space) { in Allocate()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout.h | 214 int64 memory_space() const { return memory_space_; } in memory_space() function
|
| D | layout_util.cc | 60 int64_t element_size_in_bits, int64_t memory_space) { in MakeLayout()
|
| D | xla_data.proto | 178 int64 memory_space = 8; field
|
| D | shape_util.cc | 125 int64_t element_size_in_bits, int64_t memory_space) { in MakeShapeWithLayoutInternal()
314 int64_t element_size_in_bits, int64_t memory_space) { in MakeShapeWithLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 47 int64_t memory_space) { in Allocate()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_roots_to_loops.cc | 191 auto memory_space = in isOnGpu() local
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 69 DeviceMemoryBase HostExecutor::Allocate(uint64 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()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_device.cc | 106 int64_t memory_space) override { in Allocate()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 514 DeviceMemoryBase GpuExecutor::Allocate(uint64 size, int64_t memory_space) { in Allocate()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 421 DeviceMemoryBase GpuExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate()
|