Home
last modified time | relevance | path

Searched refs:SparseTensorSpec (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor_test.py227 spec1 = sparse_tensor.SparseTensorSpec()
231 spec2 = sparse_tensor.SparseTensorSpec([None, None], dtypes.string)
236 spec1 = sparse_tensor.SparseTensorSpec()
240 (sparse_tensor.SparseTensorSpec(),
242 (sparse_tensor.SparseTensorSpec(shape=[5, None, None]),
244 (sparse_tensor.SparseTensorSpec(dtype=dtypes.int32),
255 (sparse_tensor.SparseTensorSpec(dtype=dtypes.string), [
260 (sparse_tensor.SparseTensorSpec(shape=[5, None, None]), [
271 "st_spec": sparse_tensor.SparseTensorSpec(),
277 "st_spec": sparse_tensor.SparseTensorSpec([100, None, None]),
[all …]
Dsparse_tensor.py270 return SparseTensorSpec(self.shape, self.dtype)
282 return SparseTensorSpec(tensor_shape.unknown_shape(rank), self.dtype)
296 class SparseTensorSpec(type_spec.BatchableTypeSpec): class
400 return SparseTensorSpec(
407 return SparseTensorSpec(self._shape[1:], self._dtype)
435 SparseTensor, SparseTensorSpec.from_value)
437 SparseTensorValue, SparseTensorSpec.from_value)
/external/tensorflow/tensorflow/python/data/util/
Dstructure_test.py65 sparse_tensor.SparseTensorSpec, [dtypes.variant], [None]),
553 sparse_tensor.SparseTensorSpec([2, 2], dtypes.int32)),
584 "b": (sparse_tensor.SparseTensorSpec(
627 ("SparseTensor", sparse_tensor.SparseTensorSpec([None], dtypes.float32),
628 32, sparse_tensor.SparseTensorSpec([32, None], dtypes.float32)),
630 sparse_tensor.SparseTensorSpec([4], dtypes.float32), None,
631 sparse_tensor.SparseTensorSpec([None, 4], dtypes.float32)),
641 "b": (sparse_tensor.SparseTensorSpec([2, 2], dtypes.int32),
646 "b": (sparse_tensor.SparseTensorSpec([128, 2, 2], dtypes.int32),
662 ("SparseTensor", sparse_tensor.SparseTensorSpec([32, None],
[all …]
Dstructure.py51 return sparse_tensor.SparseTensorSpec(shape, dtype)
113 if isinstance(spec, sparse_tensor.SparseTensorSpec):
169 flat_ret.append(sparse_tensor.SparseTensorSpec(flat_shape, flat_type))
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-sparse-tensor-spec.pbtxt1 path: "tensorflow.SparseTensorSpec"
3 is_instance: "<class \'tensorflow.python.framework.sparse_tensor.SparseTensorSpec\'>"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-sparse-tensor-spec.pbtxt1 path: "tensorflow.SparseTensorSpec"
3 is_instance: "<class \'tensorflow.python.framework.sparse_tensor.SparseTensorSpec\'>"
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dhashing.py213 if isinstance(input_spec, sparse_tensor.SparseTensorSpec):
214 return sparse_tensor.SparseTensorSpec(
Dcategory_crossing.py197 isinstance(inp_spec, sparse_tensor.SparseTensorSpec)
199 return sparse_tensor.SparseTensorSpec(
Ddiscretization.py226 if isinstance(input_spec, sparse_tensor.SparseTensorSpec):
227 return sparse_tensor.SparseTensorSpec(
Dcategory_encoding.py145 return sparse_tensor.SparseTensorSpec(
/external/tensorflow/tensorflow/python/keras/engine/
Dkeras_tensor_test.py54 type_spec=sparse_tensor.SparseTensorSpec(
/external/tensorflow/tensorflow/python/ops/
Dmap_fn.py542 elif isinstance(spec, sparse_tensor.SparseTensorSpec):
544 return sparse_tensor.SparseTensorSpec(None, spec.dtype)
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dparsing_ops.py72 self._element_spec[key] = sparse_tensor.SparseTensorSpec(
Dbatching.py310 self._element_spec = sparse_tensor.SparseTensorSpec(
/external/tensorflow/tensorflow/python/saved_model/
Dnested_structure_coder_test.py236 structure = [sparse_tensor.SparseTensorSpec([10, 20], dtypes.float32)]
327 "st": sparse_tensor.SparseTensorSpec([10, 20], dtypes.float32),
Dnested_structure_coder.py500 sparse_tensor.SparseTensorSpec,
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_type_spec_test.py373 sparse_tensor.SparseTensorSpec(
375 sparse_tensor.SparseTensorSpec(
Dinput_lib.py597 spec, (sparse_tensor.SparseTensorSpec, ragged_tensor.RaggedTensorSpec)):
1491 if isinstance(spec, sparse_tensor.SparseTensorSpec):
1522 if isinstance(spec, sparse_tensor.SparseTensorSpec) and spec.shape:
Dtpu_strategy.py924 if isinstance(spec, (sparse_tensor.SparseTensorSpec,
/external/tensorflow/tensorflow/core/protobuf/
Dstruct.proto129 SPARSE_TENSOR_SPEC = 1; // tf.SparseTensorSpec
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dload.py1119 if isinstance(x, sparse_tensor.SparseTensorSpec):
1120 return sparse_tensor.SparseTensorSpec(common_shape, x.dtype)
Drevive_test.py216 sparse_tensor.SparseTensorSpec):
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_generator_test.py484 output_signature=sparse_tensor.SparseTensorSpec([3, 4], dtypes.int64))
Doptional_test.py55 sparse_tensor.SparseTensorSpec([10, 10], dtypes.int32)),
Ddataset_test.py321 dense_shape=[1]), sparse_tensor.SparseTensorSpec([1], dtypes.int32))

12