Searched refs:weight_slice_size_input_values_ (Results 1 – 2 of 2) sorted by relevance
188 weight_slice_shape_, weight_slice_size_input_values_, in SetWeightForInputToCellGate()199 weight_slice_size_input_values_, in SetWeightForInputToInputGate()208 weight_slice_shape_, weight_slice_size_input_values_, in SetWeightForInputToForgetGate()218 weight_slice_shape_, weight_slice_size_input_values_, in SetWeightForInputToOutputGate()514 weight_slice_size_input_values_ = {n_cell_, n_input_}; in Initialize()
161 SmallVector<int64_t, 2> weight_slice_size_input_values_; variable