Lines Matching defs:kernel
57 // Emplace a new LstmCell operator with extended inputs (kernel/lstm.cc).
76 Array& kernel =
85 base_name + "weight_i_i", num_cell, num_input, kernel, 0, 0);
87 base_name + "weight_c_i", num_cell, num_input, kernel,
91 base_name + "weight_f_i", num_cell, num_input, kernel, num_cell * 2, 0);
94 base_name + "weight_o_i", num_cell, num_input, kernel, num_cell * 3, 0);
99 base_name + "weight_i_r", num_cell, num_output, kernel, 0, num_input);
102 base_name + "weight_c_r", num_cell, num_output, kernel,
106 base_name + "weight_f_r", num_cell, num_output, kernel,
110 base_name + "weight_o_r", num_cell, num_output, kernel,
