Searched refs:make_2d (Results 1 – 2 of 2) sorted by relevance
400 inputs_2d_tf = keras.layers.local.make_2d(inputs_tf, split_dim)520 lc_2_kernel_masked = keras.layers.local.make_2d(
735 kernel = make_2d(kernel, split_dim=K.ndim(kernel) // 2)785 def make_2d(tensor, split_dim): function