Searched refs:flat_values_spec (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_supported_values_test.py | 337 flat_values_spec = WrappedTensorSpec(
344 flat_values_spec=flat_values_spec)
349 self.assertEqual(spec1._flat_values_spec, flat_values_spec)
355 self.assertEqual(spec1.flat_values_spec, flat_values_spec)
364 flat_values_spec=flat_values_spec)
369 flat_values_spec=tensor_spec.TensorSpec(None, dtypes.float32)),
376 flat_values_spec=tensor_spec.TensorSpec(
393 flat_values_spec=WrappedTensorSpec(
398 flat_values_spec=WrappedTensorSpec(
407 flat_values_spec=WrappedTensorSpec(
[all …]
|
| D | ragged_tensor.py | 2265 def flat_values_spec(self): member in RaggedTensorSpec
2283 flat_values_spec=None): argument
2301 if flat_values_spec is not None:
2303 dtype = flat_values_spec.dtype
2304 elif dtype != flat_values_spec.dtype:
2310 self._flat_values_spec = flat_values_spec
2353 flat_values_spec = self._flat_values_spec
2354 if flat_values_spec is None:
2357 flat_values_spec = tensor_spec.TensorSpec(flat_values_shape, self._dtype)
2363 flat_values_spec,
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-ragged-tensor-spec.pbtxt | 12 name: "flat_values_spec"
33 …f\', \'shape\', \'dtype\', \'ragged_rank\', \'row_splits_dtype\', \'flat_values_spec\'], varargs=N…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-ragged-tensor-spec.pbtxt | 12 name: "flat_values_spec"
33 …f\', \'shape\', \'dtype\', \'ragged_rank\', \'row_splits_dtype\', \'flat_values_spec\'], varargs=N…
|