Searched refs:kernel_idxs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local.py | 212 self.kernel_idxs = sorted(
223 shape=(len(self.kernel_idxs),),
274 output = local_conv_sparse_matmul(inputs, self.kernel, self.kernel_idxs,
518 self.kernel_idxs = sorted(
529 shape=(len(self.kernel_idxs),),
584 output = local_conv_sparse_matmul(inputs, self.kernel, self.kernel_idxs,
744 def local_conv_sparse_matmul(inputs, kernel, kernel_idxs, kernel_shape, argument
772 a_indices=kernel_idxs,
|