Searched refs:begin_i2o_values (Results 1 – 1 of 1) sorted by relevance
215 SmallVector<int64_t, 2> begin_i2o_values = {input_output_start, 0}; in SetWeightForInputToOutputGate() local217 &builder_, weight_transposed_, weight_slice_shape_, begin_i2o_values, in SetWeightForInputToOutputGate()