Searched refs:st_input (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor.py | 1774 def from_sparse(cls, st_input, name=None, row_splits_dtype=dtypes.int64): argument
1807 if not sparse_tensor.is_sparse(st_input):
1808 raise TypeError("Expected SparseTensor, got %s" % type(st_input).__name__)
1809 with ops.name_scope(name, "RaggedFromSparse", [st_input]):
1810 st_input = sparse_tensor.convert_to_tensor_or_sparse_tensor(
1811 st_input, name="st_input")
1813 if st_input.dense_shape.shape.ndims is None:
1816 static_rank_from_dense_shape = st_input.dense_shape.shape.dims[0].value
1818 if st_input.indices.shape.ndims is None:
1821 static_rank_from_indices = st_input.indices.shape.dims[1].value
[all …]
|
| D | ragged_conversion_ops.py | 144 def from_sparse(st_input, name=None): argument
145 return ragged_tensor.RaggedTensor.from_sparse(st_input, name)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-ragged-tensor.pbtxt | 81 …argspec: "args=[\'cls\', \'st_input\', \'name\', \'row_splits_dtype\'], varargs=None, keywords=Non…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-ragged-tensor.pbtxt | 81 …argspec: "args=[\'cls\', \'st_input\', \'name\', \'row_splits_dtype\'], varargs=None, keywords=Non…
|