Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor_test.py225 spec1 = sparse_tensor.SparseTensorSpec()
229 spec2 = sparse_tensor.SparseTensorSpec([None, None], dtypes.string)
234 spec1 = sparse_tensor.SparseTensorSpec()
238 (sparse_tensor.SparseTensorSpec(),
240 (sparse_tensor.SparseTensorSpec(shape=[5, None, None]),
242 (sparse_tensor.SparseTensorSpec(dtype=dtypes.int32),
253 (sparse_tensor.SparseTensorSpec(dtype=dtypes.string), [
258 (sparse_tensor.SparseTensorSpec(shape=[5, None, None]), [
269 "st_spec": sparse_tensor.SparseTensorSpec(),
275 "st_spec": sparse_tensor.SparseTensorSpec([100, None, None]),
[all …]
Dsparse_tensor.py266 return SparseTensorSpec(self.shape, self.dtype)
279 return SparseTensorSpec(tensor_shape.unknown_shape(rank), self.dtype)
293 class SparseTensorSpec(type_spec.BatchableTypeSpec): class
398 return SparseTensorSpec(
405 return SparseTensorSpec(self._shape[1:], self._dtype)
434 SparseTensor, SparseTensorSpec.from_value)
436 SparseTensorValue, SparseTensorSpec.from_value)
/external/tensorflow/tensorflow/python/data/util/
Dstructure_test.py60 lambda: sparse_tensor.SparseTensorSpec, lambda: [dtypes.variant],
343 sparse_tensor.SparseTensorSpec([2, 2], dtypes.int32)),
377 "b": (sparse_tensor.SparseTensorSpec([2, 2], dtypes.int32),
399 (sparse_tensor.SparseTensorSpec([None], dtypes.float32), 32,
400 sparse_tensor.SparseTensorSpec([32, None], dtypes.float32)),
401 (sparse_tensor.SparseTensorSpec([4], dtypes.float32), None,
402 sparse_tensor.SparseTensorSpec([None, 4], dtypes.float32)),
410 "b": (sparse_tensor.SparseTensorSpec([2, 2], dtypes.int32),
415 "b": (sparse_tensor.SparseTensorSpec([128, 2, 2], dtypes.int32),
436 (sparse_tensor.SparseTensorSpec([32, None], dtypes.float32),
[all …]
Dstructure.py48 return sparse_tensor.SparseTensorSpec(shape, dtype)
111 if isinstance(spec, sparse_tensor.SparseTensorSpec):
170 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\'>"
Dtensorflow.pbtxt64 name: "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/ops/
Dmap_fn.py539 elif isinstance(spec, sparse_tensor.SparseTensorSpec):
541 return sparse_tensor.SparseTensorSpec(None, spec.dtype)
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_type_spec_test.py438 sparse_tensor.SparseTensorSpec(
440 sparse_tensor.SparseTensorSpec(
617 sparse_tensor.SparseTensorSpec(
619 sparse_tensor.SparseTensorSpec(
Dinput_lib.py538 spec, (sparse_tensor.SparseTensorSpec, ragged_tensor.RaggedTensorSpec)):
1510 if isinstance(spec, sparse_tensor.SparseTensorSpec):
Dtpu_strategy.py980 if isinstance(spec, (sparse_tensor.SparseTensorSpec,
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dparsing_ops.py69 self._element_spec[key] = sparse_tensor.SparseTensorSpec(
Dbatching.py315 self._element_spec = sparse_tensor.SparseTensorSpec(
/external/tensorflow/tensorflow/python/kernel_tests/
Dcomposite_tensor_ops_test.py61 sparse_tensor.SparseTensorSpec([None, None], dtypes.int32),
/external/tensorflow/tensorflow/python/saved_model/
Dnested_structure_coder_test.py235 structure = [sparse_tensor.SparseTensorSpec([10, 20], dtypes.float32)]
417 "st": sparse_tensor.SparseTensorSpec([10, 20], dtypes.float32),
Dnested_structure_coder.py468 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.py1144 if isinstance(x, sparse_tensor.SparseTensorSpec):
1145 return sparse_tensor.SparseTensorSpec(common_shape, x.dtype)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_generator_test.py479 output_signature=sparse_tensor.SparseTensorSpec([3, 4], dtypes.int64))
Doptional_test.py51 sparse_tensor.SparseTensorSpec([10, 10], dtypes.int32)),
Ddataset_test.py361 dense_shape=[1]), sparse_tensor.SparseTensorSpec([1], dtypes.int32))
Diterator_test.py838 expected_element_structure=sparse_tensor.SparseTensorSpec(
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor_test.py1077 type_spec=sparse_tensor.SparseTensorSpec([None], dtypes.int32),
/external/tensorflow/tensorflow/python/keras/
Dbackend.py982 return isinstance(spec, sparse_tensor.SparseTensorSpec)
1337 spec = sparse_tensor.SparseTensorSpec(
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py1556 [sparse_tensor.SparseTensorSpec([None], dtypes.int32)]),
5481 sparse_tensor.SparseTensorSpec([None], dtypes.int32))})

12