Home
last modified time | relevance | path

Searched refs:AllocateArray (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dtemporary_memory_manager.h62 port::StatusOr<std::unique_ptr<TemporaryDeviceMemory<T>>> AllocateArray(
137 TemporaryMemoryManager::AllocateArray(uint64 element_count) { in AllocateArray() function
Dstream_executor_pimpl.h124 DeviceMemory<T> AllocateArray(uint64 element_count, int64 memory_space = 0);
129 return ScopedDeviceMemory<T>(this, AllocateArray<T>(element_count)); in AllocateOwnedArray()
136 return AllocateArray<T>(1); in AllocateScalar()
810 inline DeviceMemory<T> StreamExecutor::AllocateArray(uint64 element_count, in AllocateArray() function
837 : ScopedDeviceMemory(parent, parent->AllocateArray<ElemT>(values.size())) { in ScopedDeviceMemory()
Dtemporary_memory_manager.cc97 stream_->parent()->AllocateArray<uint8>(byte_size); in AllocateArrayBase()
Dstream_executor_pimpl.cc902 DeviceMemoryBase result = executor->AllocateArray<uint8>(size, memory_space); in Allocate()
Dstream.h2147 return temporary_memory_manager_.AllocateArray<T>(element_count); in AllocateTemporaryArray()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinfeed_manager.h50 : device_memory_(executor, executor->AllocateArray<uint8>(length)), in InfeedBuffer()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_mem_allocator.h52 ptr = stream_exec_->AllocateArray<char>(num_bytes).opaque(); in Alloc()
/external/google-breakpad/src/client/
Dminidump_file_writer-inl.h57 inline bool TypedMDRVA<MDType>::AllocateArray(size_t count) { in AllocateArray() function
Dminidump_file_writer.h238 bool AllocateArray(size_t count);
Dminidump_file_writer_unittest.cc88 ASSERT_TRUE(array.AllocateArray(count)); in WriteFile()
/external/protobuf/src/google/protobuf/
Ddescriptor.cc628 Type* AllocateArray(int count);
1191 Type* DescriptorPool::Tables::AllocateArray(int count) { in AllocateArray() function in google::protobuf::DescriptorPool::Tables
3268 inline void AllocateArray(int size, Type** output) { in AllocateArray() function in google::protobuf::DescriptorBuilder
3269 *output = tables_->AllocateArray<Type>(size); in AllocateArray()
3906 placeholder_file->enum_types_ = tables_->AllocateArray<EnumDescriptor>(1); in NewPlaceholderWithMutexHeld()
3920 placeholder_enum->values_ = tables_->AllocateArray<EnumValueDescriptor>(1); in NewPlaceholderWithMutexHeld()
3941 placeholder_file->message_types_ = tables_->AllocateArray<Descriptor>(1); in NewPlaceholderWithMutexHeld()
3957 tables_->AllocateArray<Descriptor::ExtensionRange>(1); in NewPlaceholderWithMutexHeld()
4156 AllocateArray(INPUT.NAME##_size(), &OUTPUT->NAME##s_); \
4349 tables_->AllocateArray<const FileDescriptor*>(proto.dependency_size()); in BuildFileImpl()
[all …]
/external/google-breakpad/src/client/linux/minidump_writer/
Dminidump_writer.cc233 if (!dir.AllocateArray(kNumWriters)) in Dump()
829 if (!linkmap.AllocateArray(dso_count)) in WriteDSODebugStream()
/external/google-breakpad/src/client/solaris/handler/
Dminidump_generator.cc723 if (!dir.AllocateArray(writer_count)) in Write()
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test.cc359 DeviceMemory<int> mem = executor->AllocateArray<int>(2); in TEST_F()
/external/google-breakpad/src/client/mac/handler/
Dminidump_generator.cc256 if (!dir.AllocateArray(writer_count)) in Write()