Home
last modified time | relevance | path

Searched refs:GetBuffer (Results 1 – 17 of 17) sorted by relevance

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp38 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer() function
43 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer() function
459 const float* bwInput = GetBuffer<const float>(input_); in Eval()
463 bwInput = GetBuffer<const float>(aux_input_); in Eval()
473 fw_output_activation_state_buffer = GetBuffer<float>(fw_output_activation_state_); in Eval()
474 fw_output_cell_state_buffer = GetBuffer<float>(fw_output_cell_state_); in Eval()
486 params_, GetBuffer<const float>(input_), input_->shape(), in Eval()
487 GetBuffer<const float>(fw_input_to_input_weights_), in Eval()
488 GetBuffer<const float>(fw_input_to_forget_weights_), in Eval()
489 GetBuffer<const float>(fw_input_to_cell_weights_), in Eval()
[all …]
DLSTM.cpp38 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer() function
43 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer() function
997 LSTMEvalFloat32(params_, GetBuffer<const float>(input_), input_->shape(), in Eval()
998 GetBuffer<const float>(input_to_input_weights_), in Eval()
999 GetBuffer<const float>(input_to_forget_weights_), in Eval()
1000 GetBuffer<const float>(input_to_cell_weights_), in Eval()
1001 GetBuffer<const float>(input_to_output_weights_), in Eval()
1003 GetBuffer<const float>(recurrent_to_input_weights_), in Eval()
1004 GetBuffer<const float>(recurrent_to_forget_weights_), in Eval()
1005 GetBuffer<const float>(recurrent_to_cell_weights_), in Eval()
[all …]
DMultinomial.cpp43 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer() function
48 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer() function
89 convertFloat16ToFloat32(GetBuffer<_Float16>(input_), &inputDataFloat32); in Eval()
94 EvalFloat32(GetBuffer<float>(input_)); in Eval()
110 int32_t* seeds = GetBuffer<int32_t>(random_seeds_); in EvalFloat32()
140 auto* output_ptr_batch = GetBuffer<int32_t>(output_) + b * sample_count_; in EvalFloat32()
DQuantizedLSTM.cpp40 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer() function
45 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer() function
212 const uint8_t* submatrixValues = GetBuffer<uint8_t>(submatrix); in assignWeightsSubmatrix()
369 memcpy(bias + 0 * outputSize, GetBuffer<int32_t>(inputGateBias_), sizeof(int32_t) * outputSize); in concatenateBiases()
370 memcpy(bias + 1 * outputSize, GetBuffer<int32_t>(cellGateBias_), sizeof(int32_t) * outputSize); in concatenateBiases()
371 memcpy(bias + 2 * outputSize, GetBuffer<int32_t>(forgetGateBias_), in concatenateBiases()
373 memcpy(bias + 3 * outputSize, GetBuffer<int32_t>(outputGateBias_), in concatenateBiases()
415 GetBuffer<const uint8_t>(input_), convertShapeToDims(input_->shape()), in eval()
416 GetBuffer<const uint8_t>(prevOutput_), convertShapeToDims(prevOutput_->shape()), in eval()
418 convertShapeToDims(biasShape), GetBuffer<const int16_t>(prevCellState_), in eval()
[all …]
/frameworks/native/libs/vr/libpdx/
Dthread_local_buffer_tests.cpp98 MessageBuffer<SendSlotX>::GetBuffer(); in TEST()
108 auto& buffer1 = MessageBuffer<SendSlotA>::GetBuffer(); in TEST()
109 auto& buffer2 = MessageBuffer<SendSlotA>::GetBuffer(); in TEST()
110 auto& buffer3 = MessageBuffer<SendSlotB>::GetBuffer(); in TEST()
111 auto& buffer4 = MessageBuffer<SendSlotB>::GetBuffer(); in TEST()
/frameworks/native/services/vr/bufferhubd/
Dconsumer_channel.cpp73 case BufferHubRPC::GetBuffer::Opcode: in HandleMessage()
74 DispatchRemoteMethod<BufferHubRPC::GetBuffer>( in HandleMessage()
103 return {producer->GetBuffer(client_state_mask_)}; in OnGetBuffer()
Dproducer_channel.cpp203 case BufferHubRPC::GetBuffer::Opcode: in HandleMessage()
204 DispatchRemoteMethod<BufferHubRPC::GetBuffer>( in HandleMessage()
228 BufferDescription<BorrowedHandle> ProducerChannel::GetBuffer( in GetBuffer() function in android::dvr::ProducerChannel
244 return {GetBuffer(BufferHubDefs::kFirstClientBitMask)}; in OnGetBuffer()
/frameworks/native/libs/vr/libbufferhubqueue/include/private/dvr/
Dbuffer_hub_queue_client.h97 std::shared_ptr<BufferHubBase> GetBuffer(size_t slot) const { in GetBuffer() function
314 std::shared_ptr<ProducerBuffer> GetBuffer(size_t slot) const { in GetBuffer() function
316 BufferHubQueue::GetBuffer(slot)); in GetBuffer()
419 std::shared_ptr<ConsumerBuffer> GetBuffer(size_t slot) const { in GetBuffer() function
421 BufferHubQueue::GetBuffer(slot)); in GetBuffer()
/frameworks/native/libs/vr/libvrflinger/
Dhardware_composer.h124 IonBuffer* GetBuffer();
228 IonBuffer* GetBuffer() { in GetBuffer() function
263 IonBuffer* GetBuffer() { return buffer.get(); } in GetBuffer() function
Dhardware_composer.cpp1385 IonBuffer* Layer::GetBuffer() { in GetBuffer() function in android::dvr::Layer
1387 IonBuffer* operator()(SourceSurface& source) { return source.GetBuffer(); } in GetBuffer()
1388 IonBuffer* operator()(SourceBuffer& source) { return source.GetBuffer(); } in GetBuffer()
/frameworks/native/libs/vr/libpdx/private/pdx/rpc/
Dthread_local_buffer.h67 static BufferType& GetBuffer(std::size_t capacity = Capacity) {
/frameworks/native/services/vr/bufferhubd/include/private/dvr/
Dproducer_channel.h54 BufferDescription<BorrowedHandle> GetBuffer(uint32_t client_state_mask);
/frameworks/native/libs/vr/libbufferhub/
Dbuffer_hub_base.cpp49 InvokeRemoteMethod<BufferHubRPC::GetBuffer>(); in ImportBuffer()
/frameworks/native/libs/gui/
DBufferHubProducer.cpp252 std::shared_ptr<ProducerBuffer> producer_buffer = queue_->GetBuffer(slot); in DetachBufferLocked()
794 auto producer_buffer = queue_->GetBuffer(slot); in AllocateBuffer()
/frameworks/native/libs/vr/libbufferhub/include/private/dvr/
Dbufferhub_rpc.h335 PDX_REMOTE_METHOD(GetBuffer, kOpGetBuffer,
/frameworks/native/vulkan/api/
Dvulkan.api8273 bufferObject := GetBuffer(buffer)
8283 bufferObject := GetBuffer(buffer)
8621 bufferObject := GetBuffer(buffer)
8640 bufferObject := GetBuffer(pCreateInfo.buffer)
9426 bufferObject := GetBuffer(buffer)
9449 bufferObject := GetBuffer(buffer)
9491 bufferObject := GetBuffer(buffer)
9507 bufferObject := GetBuffer(buffer)
9532 bufferObject := GetBuffer(buffer)
9548 srcBufferObject := GetBuffer(srcBuffer)
[all …]
/frameworks/native/libs/vr/libbufferhubqueue/
Dbuffer_hub_queue_client.cpp635 std::shared_ptr<ProducerBuffer> buffer = ProducerQueue::GetBuffer(*iter); in DequeueUnacquiredBuffer()