HomeSort by relevance Sort by last modified time
    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