Searched refs:SliceRankedTensor (Results 1 – 1 of 1) sorted by relevance
128 Value SliceRankedTensor(OpBuilder* builder, Value input, in SliceRankedTensor() function186 input2cell_ = SliceRankedTensor( in SetWeightForInputToCellGate()196 : SliceRankedTensor(&builder_, weight_transposed_, in SetWeightForInputToInputGate()206 input2forget_ = SliceRankedTensor( in SetWeightForInputToForgetGate()216 input2output_ = SliceRankedTensor( in SetWeightForInputToOutputGate()224 rec2cell_ = SliceRankedTensor( in SetWeightForRecurrentToCellGate()234 : SliceRankedTensor(&builder_, weight_transposed_, in SetWeightForRecurrentToInputGate()244 rec2forget_ = SliceRankedTensor( in SetWeightForRecurrentToForgetGate()253 rec2output_ = SliceRankedTensor( in SetWeightForRecurrentToOutputGate()261 bias2cell_ = SliceRankedTensor(&builder_, bias_, bias_slice_shape_, in SetBiasToCellGate()[all …]