OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:kCellToInputWeightsTensor
(Results
1 - 7
of
7
) sorted by null
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
lstm_utils.h
38
kCellToInputWeightsTensor
= 9, // Optional
identify_lstm_split_inputs.cc
114
CreateOptionalArray(model, &(lstm_cell_op->inputs[
kCellToInputWeightsTensor
]),
/frameworks/ml/nn/common/operations/
LSTM.h
77
static constexpr int
kCellToInputWeightsTensor
= 9; // Optional
LSTM.cpp
54
cell_to_input_weights_ = GetInput(operation, operands,
kCellToInputWeightsTensor
); // optional
147
GetInput(operation, operands, LSTMCell::
kCellToInputWeightsTensor
);
LSTMTest.cpp
219
execution.setInput(LSTMCell::
kCellToInputWeightsTensor
, nullptr, 0);
222
execution.setInput(LSTMCell::
kCellToInputWeightsTensor
, nullptr, 0);
[
all
...]
/external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc
52
constexpr int
kCellToInputWeightsTensor
= 9; // Optional
140
GetOptionalInputTensor(context, node,
kCellToInputWeightsTensor
);
329
GetOptionalInputTensor(context, node,
kCellToInputWeightsTensor
);
unidirectional_sequence_lstm.cc
52
constexpr int
kCellToInputWeightsTensor
= 9; // Optional
140
GetOptionalInputTensor(context, node,
kCellToInputWeightsTensor
);
331
GetOptionalInputTensor(context, node,
kCellToInputWeightsTensor
);
Completed in 201 milliseconds