OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:kInputToInputWeightsTensor
(Results
1 - 8
of
8
) sorted by null
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
identify_lstm_merge_inputs.cc
56
int num_cell = model->GetArray(src_op->inputs[
kInputToInputWeightsTensor
])
59
int num_input = model->GetArray(src_op->inputs[
kInputToInputWeightsTensor
])
86
model->GetArray(src_op->inputs[
kInputToInputWeightsTensor
]), 0, 0);
168
DeleteArrayIfUnused(src_op->inputs[
kInputToInputWeightsTensor
], model);
lstm_utils.h
30
kInputToInputWeightsTensor
= 1, // Optional
identify_lstm_split_inputs.cc
84
model, &(lstm_cell_op->inputs[
kInputToInputWeightsTensor
]),
/frameworks/ml/nn/common/operations/
LSTM.h
65
static constexpr int
kInputToInputWeightsTensor
= 1; // Optional
LSTM.cpp
43
input_to_input_weights_ = GetInput(operation, operands,
kInputToInputWeightsTensor
); // optional
98
GetInput(operation, operands, LSTMCell::
kInputToInputWeightsTensor
);
286
GetInput(operation, operands, LSTMCell::
kInputToInputWeightsTensor
);
LSTMTest.cpp
213
execution.setInput(LSTMCell::
kInputToInputWeightsTensor
, nullptr, 0);
[
all
...]
/external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc
40
constexpr int
kInputToInputWeightsTensor
= 1; // Optional
86
GetOptionalInputTensor(context, node,
kInputToInputWeightsTensor
);
285
GetOptionalInputTensor(context, node,
kInputToInputWeightsTensor
);
311
GetOptionalInputTensor(context, node,
kInputToInputWeightsTensor
);
unidirectional_sequence_lstm.cc
40
constexpr int
kInputToInputWeightsTensor
= 1; // Optional
86
GetOptionalInputTensor(context, node,
kInputToInputWeightsTensor
);
287
GetOptionalInputTensor(context, node,
kInputToInputWeightsTensor
);
313
GetOptionalInputTensor(context, node,
kInputToInputWeightsTensor
);
Completed in 227 milliseconds