Searched refs:sp_inputs (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_concat_op.cc | 70 std::vector<sparse::SparseTensor> sp_inputs; in operator ()() local
78 sp_inputs.push_back(std::move(tensor)); in operator ()()
79 sp_inputs[i].Reorder<T>(concat_order); in operator ()()
82 sparse::SparseTensor concat = sparse::SparseTensor::Concat<T>(sp_inputs); in operator ()()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 71 def _convert_to_sparse_tensors(sp_inputs): argument
85 if isinstance(sp_inputs, list):
86 return [_convert_to_sparse_tensor(sp_input) for sp_input in sp_inputs]
87 if isinstance(sp_inputs, tuple):
88 return (_convert_to_sparse_tensor(sp_input) for sp_input in sp_inputs)
280 sp_inputs, argument
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… argument
393 sp_inputs = _convert_to_sparse_tensors(sp_inputs)
395 if len(sp_inputs) == 1: # Degenerate case of one tensor.
[all …]
|
| D | sparse_ops_test.py | 292 result = sparse_ops.sparse_concat_v2(axis=1, sp_inputs=[input_a, input_b])
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseConcat.pbtxt | 67 sp_inputs[0]: shape = [2, 3]
72 sp_inputs[1]: shape = [2, 4]
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_concat_op_test.py | 376 sp_inputs = [
384 sp_concat = sparse_ops.sparse_concat(concat_dim, sp_inputs)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.sparse.pbtxt | 17 …argspec: "args=[\'axis\', \'sp_inputs\', \'expand_nonconcat_dims\', \'name\'], varargs=None, keywo…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.sparse.pbtxt | 21 …argspec: "args=[\'axis\', \'sp_inputs\', \'name\', \'expand_nonconcat_dim\', \'concat_dim\', \'exp…
|
| D | tensorflow.pbtxt | 2161 …argspec: "args=[\'axis\', \'sp_inputs\', \'name\', \'expand_nonconcat_dim\', \'concat_dim\', \'exp…
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distributed_training_utils_v1.py | 1145 return sparse_ops.sparse_concat_v2(axis=0, sp_inputs=outputs)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_utils_v1.py | 253 return sparse_ops.sparse_concat(sp_inputs=[target, to_append], axis=0)
|
| D | training.py | 2847 return sparse_ops.sparse_concat_v2(axis=axis, sp_inputs=tensors)
|