Searched refs:sparse_concat_v2 (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | sparse_ops.py | 388 return sparse_concat_v2(axis, sp_inputs, expand_nonconcat_dim, name)
392 def sparse_concat_v2(axis, sp_inputs, expand_nonconcat_dims=False, name=None): # pylint: disable=m… function
445 sparse_concat_v2.__doc__ = sparse_concat.__doc__.replace(
|
| H A D | sparse_ops_test.py | 292 result = sparse_ops.sparse_concat_v2(axis=1, sp_inputs=[input_a, input_b])
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A D | distributed_training_utils_v1.py | 1145 return sparse_ops.sparse_concat_v2(axis=0, sp_inputs=outputs)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training.py | 2847 return sparse_ops.sparse_concat_v2(axis=axis, sp_inputs=tensors)
|