Home
last modified time | relevance | path

Searched refs:sp_inputs (Results 1 – 10 of 10) 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.py72 def _convert_to_sparse_tensors(sp_inputs): argument
86 if isinstance(sp_inputs, list):
87 return [_convert_to_sparse_tensor(sp_input) for sp_input in sp_inputs]
88 if isinstance(sp_inputs, tuple):
89 return (_convert_to_sparse_tensor(sp_input) for sp_input in sp_inputs)
197 sp_inputs, argument
305 return sparse_concat_v2(axis, sp_inputs, expand_nonconcat_dim, name)
309 def sparse_concat_v2(axis, sp_inputs, expand_nonconcat_dims=False, name=None): # pylint: disable=m… argument
310 sp_inputs = _convert_to_sparse_tensors(sp_inputs)
312 if len(sp_inputs) == 1: # Degenerate case of one tensor.
[all …]
/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/contrib/factorization/python/ops/
Dwals_test.py66 sp_inputs = self.np_array_to_sparse(self.INPUT_MATRIX)
68 sp_inputs, current_rows, current_cols, self._regularization_coeff,
Dfactorization_ops_test.py53 def calculate_loss_from_wals_model(self, wals_model, sp_inputs): argument
71 sp_inputs, current_rows, current_cols, wals_model._regularization,
/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.pbtxt17 …argspec: "args=[\'axis\', \'sp_inputs\', \'name\', \'expand_nonconcat_dim\', \'concat_dim\', \'exp…
Dtensorflow.pbtxt2069 …argspec: "args=[\'axis\', \'sp_inputs\', \'name\', \'expand_nonconcat_dim\', \'concat_dim\', \'exp…
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dfeature_column.py600 return sparse_ops.sparse_concat(axis=1, sp_inputs=result, name="lookup")