| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 38 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 …]
|
| D | LSTM.cpp | 38 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 …]
|
| D | Multinomial.cpp | 43 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()
|
| D | QuantizedLSTM.cpp | 40 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/ |
| D | thread_local_buffer_tests.cpp | 98 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/ |
| D | consumer_channel.cpp | 73 case BufferHubRPC::GetBuffer::Opcode: in HandleMessage()
74 DispatchRemoteMethod<BufferHubRPC::GetBuffer>( in HandleMessage()
103 return {producer->GetBuffer(client_state_mask_)}; in OnGetBuffer()
|
| D | producer_channel.cpp | 203 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/ |
| D | buffer_hub_queue_client.h | 97 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/ |
| D | hardware_composer.h | 125 IonBuffer* GetBuffer();
229 IonBuffer* GetBuffer() { in GetBuffer() function
264 IonBuffer* GetBuffer() { return buffer.get(); } in GetBuffer() function
|
| D | hardware_composer.cpp | 1413 IonBuffer* Layer::GetBuffer() { in GetBuffer() function in android::dvr::Layer
1415 IonBuffer* operator()(SourceSurface& source) { return source.GetBuffer(); } in GetBuffer()
1416 IonBuffer* operator()(SourceBuffer& source) { return source.GetBuffer(); } in GetBuffer()
|
| /frameworks/native/libs/vr/libpdx/private/pdx/rpc/ |
| D | thread_local_buffer.h | 67 static BufferType& GetBuffer(std::size_t capacity = Capacity) {
|
| /frameworks/native/services/vr/bufferhubd/include/private/dvr/ |
| D | producer_channel.h | 54 BufferDescription<BorrowedHandle> GetBuffer(uint32_t client_state_mask);
|
| /frameworks/native/libs/vr/libbufferhub/ |
| D | buffer_hub_base.cpp | 49 InvokeRemoteMethod<BufferHubRPC::GetBuffer>(); in ImportBuffer()
|
| /frameworks/native/libs/gui/ |
| D | BufferHubProducer.cpp | 251 std::shared_ptr<ProducerBuffer> producer_buffer = queue_->GetBuffer(slot); in DetachBufferLocked()
793 auto producer_buffer = queue_->GetBuffer(slot); in AllocateBuffer()
|
| /frameworks/native/libs/vr/libbufferhub/include/private/dvr/ |
| D | bufferhub_rpc.h | 335 PDX_REMOTE_METHOD(GetBuffer, kOpGetBuffer,
|
| /frameworks/native/libs/vr/libbufferhubqueue/ |
| D | buffer_hub_queue_client.cpp | 635 std::shared_ptr<ProducerBuffer> buffer = ProducerQueue::GetBuffer(*iter); in DequeueUnacquiredBuffer()
|