OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:SetInput
(Results
1 - 4
of
4
) sorted by null
/frameworks/ml/nn/common/operations/
RNNTest.cpp
198
void
SetInput
(int offset, float* begin, float* end) {
224
ASSERT_EQ(execution.
setInput
(RNN::k##X##Tensor, X##_.data(), \
241
ASSERT_EQ(execution.
setInput
(RNN::kActivationParam, &activation_,
320
rnn.
SetInput
(0, batch_start, batch_end);
321
rnn.
SetInput
(rnn.input_size(), batch_start, batch_end);
LSTMTest.cpp
174
void
SetInput
(int offset, float *begin, float *end) {
195
ASSERT_EQ(execution.
setInput
(LSTMCell::k##X##Tensor, X##_.data(), \
213
execution.
setInput
(LSTMCell::kInputToInputWeightsTensor, nullptr, 0);
214
execution.
setInput
(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0);
219
execution.
setInput
(LSTMCell::kCellToInputWeightsTensor, nullptr, 0);
222
execution.
setInput
(LSTMCell::kCellToInputWeightsTensor, nullptr, 0);
223
execution.
setInput
(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0);
224
execution.
setInput
(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0);
229
execution.
setInput
(LSTMCell::kProjectionBiasTensor, nullptr, 0);
232
execution.
setInput
(LSTMCell::kProjectionWeightsTensor, nullptr, 0)
[
all
...]
SVDFTest.cpp
200
ASSERT_EQ(execution.
setInput
(SVDF::k##X##Tensor, X##_.data(), \
218
ASSERT_EQ(execution.
setInput
(SVDF::kRankParam, &rank_, sizeof(rank_)),
222
ASSERT_EQ(execution.
setInput
(SVDF::kActivationParam, &activation,
238
void
SetInput
(int offset, float* begin, float* end) {
307
svdf.
SetInput
(0, batch_start, batch_end);
/external/v8/src/compiler/
instruction.cc
610
void PhiInstruction::
SetInput
(size_t offset, int virtual_register) {
[
all
...]
Completed in 449 milliseconds