Home
last modified time | relevance | path

Searched refs:sp_inputs (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_concat_op.cc124 std::vector<sparse::SparseTensor> sp_inputs; in Compute() local
132 sp_inputs.push_back(std::move(tensor)); in Compute()
133 sp_inputs[i].Reorder<T>(concat_order); in Compute()
136 sparse::SparseTensor concat = sparse::SparseTensor::Concat<T>(sp_inputs); in Compute()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py76 def _convert_to_sparse_tensors(sp_inputs): argument
90 if isinstance(sp_inputs, list):
91 return [_convert_to_sparse_tensor(sp_input) for sp_input in sp_inputs]
92 if isinstance(sp_inputs, tuple):
93 return (_convert_to_sparse_tensor(sp_input) for sp_input in sp_inputs)
285 sp_inputs, argument
393 return sparse_concat_v2(axis, sp_inputs, expand_nonconcat_dim, name)
397 def sparse_concat_v2(axis, sp_inputs, expand_nonconcat_dims=False, name=None): # pylint: disable=m… argument
398 sp_inputs = _convert_to_sparse_tensors(sp_inputs)
400 if len(sp_inputs) == 1: # Degenerate case of one tensor.
[all …]
Dsparse_ops_test.py296 result = sparse_ops.sparse_concat_v2(axis=1, sp_inputs=[input_a, input_b])
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseConcat.pbtxt67 sp_inputs[0]: shape = [2, 3]
72 sp_inputs[1]: shape = [2, 4]
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_concat_op_test.py331 sp_inputs = [
339 sp_concat = sparse_ops.sparse_concat(concat_dim, sp_inputs)
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dcategory_crossing.py170 return sparse_ops.sparse_concat_v2(axis=1, sp_inputs=outputs)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.sparse.pbtxt17 …argspec: "args=[\'axis\', \'sp_inputs\', \'expand_nonconcat_dims\', \'name\'], varargs=None, keywo…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.sparse.pbtxt21 …argspec: "args=[\'axis\', \'sp_inputs\', \'name\', \'expand_nonconcat_dim\', \'concat_dim\', \'exp…
Dtensorflow.pbtxt2153 …argspec: "args=[\'axis\', \'sp_inputs\', \'name\', \'expand_nonconcat_dim\', \'concat_dim\', \'exp…
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistributed_training_utils_v1.py1155 return sparse_ops.sparse_concat_v2(axis=0, sp_inputs=outputs)
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_v1.py259 return sparse_ops.sparse_concat(sp_inputs=[target, to_append], axis=0)
Dtraining.py2771 return sparse_ops.sparse_concat_v2(axis=axis, sp_inputs=tensors)