Searched refs:row_axis (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/layers/python/ops/ |
| D | sparse_ops.py | 190 def sparse_row_envelope(sparse_input, row_axis=0, col_axis=1, name=None): argument
212 if not (isinstance(row_axis, compat.integral_types) and
216 if row_axis == col_axis:
221 row_indices = indices[:, row_axis]
223 num_rows = math_ops.cast(sparse_input.dense_shape[row_axis], dtypes.int32)
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | sugar_test.py | 60 self.row_axis = ('row', range(self.num_rows))
66 self.batch_axis, self.row_axis, self.column_axis, self.channel_axis,
83 self.batch_axis, self.row_axis, self.column_axis,
98 self.batch_axis, self.row_axis, self.column_axis, depth_axis,
|