Searched defs:input_layer_size (Results 1 – 2 of 2) sorted by relevance
135 int64_t input_layer_size = layer == 0 ? input_size : hidden_size * num_directions; in LstmGradWeightInferShape() local
209 int64_t input_layer_size = input_size; in GetWeightSize() local