Home
last modified time | relevance | path

Searched refs:SparseTensorStructure (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/data/util/
Dstructure_test.py49 structure.SparseTensorStructure, [dtypes.variant], [None]),
345 structure.SparseTensorStructure(dtypes.int32, [2, 2])),
353 "b": (structure.SparseTensorStructure(dtypes.int32, [2, 2]),
400 ("SparseTensor", structure.SparseTensorStructure(dtypes.float32, [None]),
401 32, structure.SparseTensorStructure(dtypes.float32, [32, None])),
403 structure.SparseTensorStructure(dtypes.float32, [4]), None,
404 structure.SparseTensorStructure(dtypes.float32, [None, 4])),
407 "b": (structure.SparseTensorStructure(dtypes.int32, [2, 2]),
411 "b": (structure.SparseTensorStructure(dtypes.int32, [128, 2, 2]),
428 structure.SparseTensorStructure(dtypes.float32, [32, None]),
[all …]
Dstructure.py205 return SparseTensorStructure.from_value(value)
280 flat_ret.append(SparseTensorStructure(flat_type, flat_shape))
501 class SparseTensorStructure(Structure): class
521 return (isinstance(other, SparseTensorStructure) and
552 return SparseTensorStructure(
566 return SparseTensorStructure(
573 return SparseTensorStructure(self._dtype, self._dense_shape[1:])
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-sparse-tensor-structure.pbtxt1 path: "tensorflow.data.experimental.SparseTensorStructure"
3 is_instance: "<class \'tensorflow.python.data.util.structure.SparseTensorStructure\'>"
Dtensorflow.data.experimental.pbtxt48 name: "SparseTensorStructure"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.-sparse-tensor-structure.pbtxt1 path: "tensorflow.data.experimental.SparseTensorStructure"
3 is_instance: "<class \'tensorflow.python.data.util.structure.SparseTensorStructure\'>"
Dtensorflow.data.experimental.pbtxt48 name: "SparseTensorStructure"
/external/tensorflow/tensorflow/python/data/experimental/
D__init__.py130 from tensorflow.python.data.util.structure import SparseTensorStructure
/external/tensorflow/tensorflow/python/data/kernel_tests/
Ddataset_test.py220 structure.SparseTensorStructure(dtypes.int32, [1])),
Doptional_test.py260 structure.SparseTensorStructure(dtypes.int32, [1])),
Diterator_test.py852 structure.SparseTensorStructure(dtypes.int32, [1]),
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dbatching.py461 self._structure = structure.SparseTensorStructure(