OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:kInputGateBiasTensor
(Results
1 - 7
of
7
) sorted by null
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
lstm_utils.h
41
kInputGateBiasTensor
= 12, // Optional
identify_lstm_merge_inputs.cc
125
model->GetArray(src_op->inputs[
kInputGateBiasTensor
]), 0,
176
DeleteArrayIfUnused(src_op->inputs[
kInputGateBiasTensor
], model);
identify_lstm_split_inputs.cc
126
CopySubArrayToArray(model, &(lstm_cell_op->inputs[
kInputGateBiasTensor
]),
/frameworks/ml/nn/common/operations/
LSTM.h
82
static constexpr int
kInputGateBiasTensor
= 12; // Optional
LSTM.cpp
58
input_gate_bias_ = GetInput(operation, operands,
kInputGateBiasTensor
);
180
GetInput(operation, operands, LSTMCell::
kInputGateBiasTensor
);
/external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc
57
constexpr int
kInputGateBiasTensor
= 12; // Optional
173
GetOptionalInputTensor(context, node,
kInputGateBiasTensor
);
336
GetOptionalInputTensor(context, node,
kInputGateBiasTensor
);
unidirectional_sequence_lstm.cc
57
constexpr int
kInputGateBiasTensor
= 12; // Optional
173
GetOptionalInputTensor(context, node,
kInputGateBiasTensor
);
338
GetOptionalInputTensor(context, node,
kInputGateBiasTensor
);
Completed in 106 milliseconds