Home
last modified time | relevance | path

Searched refs:scratch_ (Results 1 – 14 of 14) sorted by relevance

/external/protobuf/src/google/protobuf/stubs/
Dsubstitute.h105 : text_(scratch_), size_(1) { scratch_[0] = value; } in SubstituteArg()
107 : text_(FastInt32ToBuffer(value, scratch_)), size_(strlen(text_)) {} in SubstituteArg()
109 : text_(FastUInt32ToBuffer(value, scratch_)), size_(strlen(text_)) {} in SubstituteArg()
111 : text_(FastInt32ToBuffer(value, scratch_)), size_(strlen(text_)) {} in SubstituteArg()
113 : text_(FastUInt32ToBuffer(value, scratch_)), size_(strlen(text_)) {} in SubstituteArg()
115 : text_(FastLongToBuffer(value, scratch_)), size_(strlen(text_)) {} in SubstituteArg()
117 : text_(FastULongToBuffer(value, scratch_)), size_(strlen(text_)) {} in SubstituteArg()
119 : text_(FastInt64ToBuffer(value, scratch_)), size_(strlen(text_)) {} in SubstituteArg()
121 : text_(FastUInt64ToBuffer(value, scratch_)), size_(strlen(text_)) {} in SubstituteArg()
123 : text_(FloatToBuffer(value, scratch_)), size_(strlen(text_)) {} in SubstituteArg()
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceCuda.h102 CudaStreamDevice() : stream_(&default_stream), scratch_(NULL), semaphore_(NULL) { in initializeDeviceProp()
107 …CudaStreamDevice(int device) : stream_(&default_stream), device_(device), scratch_(NULL), semaphor… in initializeDeviceProp()
115 : stream_(stream), device_(device), scratch_(NULL), semaphore_(NULL) { in initializeDeviceProp()
130 if (scratch_) { in initializeDeviceProp()
131 deallocate(scratch_); in initializeDeviceProp()
159 if (scratch_ == NULL) { in initializeDeviceProp()
160 scratch_ = allocate(kCudaScratchSize + sizeof(unsigned int)); in initializeDeviceProp()
162 return scratch_; in initializeDeviceProp()
179 mutable void* scratch_; in initializeDeviceProp() variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc83 : scratch_(scratch) {} in ScratchBufAllocator()
88 return scratch_.size(); in GetMemoryLimitInBytes()
97 if (byte_size > scratch_.size()) { in AllocateBytes()
100 " bytes from a ScratchBufAllocator of size ", scratch_.size())); in AllocateBytes()
104 return se::DeviceMemory<uint8>(scratch_); in AllocateBytes()
108 se::DeviceMemoryBase scratch_; member in xla::gpu::__anon2b86fa710111::ScratchBufAllocator
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.h51 scratch_(nullptr), in ROCMFftPlan()
85 DeviceMemory<uint8> scratch_; variable
Drocm_fft.cc265 if (!allocated.ok() || (scratch_ = allocated.ValueOrDie()) == nullptr) { in Initialize()
271 ret = wrap::hipfftSetWorkArea(parent, plan_, scratch_.opaque()); in Initialize()
321 if (!allocated.ok() || (scratch_ = allocated.ValueOrDie()) == nullptr) { in Initialize()
327 ret = wrap::hipfftSetWorkArea(parent, plan_, scratch_.opaque()); in Initialize()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.h51 scratch_(nullptr), in CUDAFftPlan()
89 DeviceMemory<uint8> scratch_; variable
Dcuda_fft.cc249 if (!allocated.ok() || (scratch_ = allocated.ValueOrDie()) == nullptr) { in UpdateScratchAllocator()
256 cufftResult_t ret = cufftSetWorkArea(plan_, scratch_.opaque()); in UpdateScratchAllocator()
/external/flatbuffers/include/flatbuffers/
Dflatbuffers.h656 scratch_(nullptr) {}
672 scratch_(other.scratch_) {
680 other.scratch_ = nullptr;
717 scratch_ = buf_;
759 FLATBUFFERS_ASSERT(cur_ >= scratch_ && scratch_ >= buf_);
760 if (len > static_cast<size_t>(cur_ - scratch_)) { reallocate(len); }
781 return static_cast<uoffset_t>(scratch_ - buf_);
797 FLATBUFFERS_ASSERT(scratch_);
798 return scratch_;
815 *reinterpret_cast<T *>(scratch_) = t;
[all …]
/external/libaom/libaom/av1/encoder/
Dmathutils.h68 double *scratch_ = NULL; in least_squares() local
71 scratch_ = (double *)aom_malloc(sizeof(*scratch) * n * (n + 1)); in least_squares()
72 scratch = scratch_; in least_squares()
88 if (scratch_) aom_free(scratch_); in least_squares()
/external/tensorflow/tensorflow/core/kernels/
Dmeta_support.cc47 Scratch() : scratch_(new uint8_t[kScratchSize]) { in Scratch()
51 scratch_.get() + kAlignment - in Scratch()
52 (reinterpret_cast<uintptr_t>(scratch_.get()) % kAlignment); in Scratch()
60 std::unique_ptr<uint8_t> scratch_; member in tensorflow::meta::__anonebe48a300111::Scratch
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc116 : scratch_(nullptr), semaphore_(nullptr), context_(nullptr) { in EigenGpuStreamDevice()
128 scratch_ = scratch; in Reinitialize()
180 void* scratchpad() const override { return scratch_; } in scratchpad()
219 mutable char* scratch_; in asyncFree() member in tensorflow::EigenGpuStreamDevice
336 for (auto sb : scratch_) gpu_allocator_->DeallocateRaw(sb); in ~BaseGPUDevice()
343 if (scratch_.size() < max_streams_) { in InitScratchBuffers()
346 if (scratch_.size() > i && scratch_[i]) continue; in InitScratchBuffers()
366 scratch_.push_back(static_cast<char*>(scratch_buffer)); in InitScratchBuffers()
939 scratch_[stream_id]); in ReinitializeDevice()
Dgpu_device.h148 gtl::InlinedVector<char*, 4> scratch_; variable
/external/tensorflow/tensorflow/core/platform/
Denv.cc476 Status s = file_->Read(pos_, kBufSize, &result, scratch_); in Next()
493 char scratch_[kBufSize]; member in tensorflow::__anonc655bce70111::FileStream
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Target/AMDGPU/
DAMDGPU.td57 "Have scratch_* flat memory instructions"