Home
last modified time | relevance | path

Searched refs:make_2d (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/
Dlocal.py699 kernel = make_2d(kernel, split_dim=K.ndim(kernel) // 2)
707 def make_2d(tensor, split_dim): function
Dlocal_test.py332 inputs_2d_tf = keras.layers.local.make_2d(inputs_tf, split_dim)