Home
last modified time | relevance | path

Searched refs:RaggedTensorSpec (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_supported_values_test.py41 from tensorflow.python.ops.ragged.ragged_tensor import RaggedTensorSpec
339 spec1 = RaggedTensorSpec(
359 spec1 = RaggedTensorSpec(
367 (RaggedTensorSpec(
372 (RaggedTensorSpec(
390 (RaggedTensorSpec(
396 (RaggedTensorSpec(
404 (RaggedTensorSpec(
414 (RaggedTensorSpec(
431 RaggedTensorSpec(
[all …]
Dragged_tensor_test.py49 from tensorflow.python.ops.ragged.ragged_tensor import RaggedTensorSpec
1706 spec1 = RaggedTensorSpec(ragged_rank=1)
1717 spec2 = RaggedTensorSpec(shape=[None, None, None])
1724 RaggedTensorSpec()
1726 RaggedTensorSpec(ragged_rank=constant_op.constant(1))
1729 RaggedTensorSpec(ragged_rank=2, shape=[None, None])
1732 spec1 = RaggedTensorSpec(ragged_rank=1)
1734 spec2 = RaggedTensorSpec(ragged_rank=0)
1738 (RaggedTensorSpec(ragged_rank=1),
1740 (RaggedTensorSpec(shape=[5, None, None]),
[all …]
Dragged_map_ops.py179 return ragged_tensor.RaggedTensorSpec(
Dragged_tensor.py2114 return RaggedTensorSpec.from_value(self)
2117 return RaggedTensorSpec(shape, self.dtype, self.ragged_rank,
2182 class RaggedTensorSpec(type_spec.BatchableTypeSpec): class
2333 elif not isinstance(spec_or_value, (RaggedTensor, RaggedTensorSpec)):
2335 return super(RaggedTensorSpec, self).is_compatible_with(spec_or_value)
2457 return RaggedTensorSpec(
2468 return RaggedTensorSpec(self._shape[1:], self._dtype, self._ragged_rank - 1,
2499 ragged_tensor_value.RaggedTensorValue, RaggedTensorSpec.from_value)
Dragged_print_op_test.py184 input_signature=[ragged_tensor.RaggedTensorSpec(ragged_rank=1)])
/external/tensorflow/tensorflow/python/data/util/
Dstructure_test.py67 ragged_tensor.RaggedTensorSpec, [dtypes.variant], [None]),
246 ragged_tensor.RaggedTensorSpec(None, dtypes.int32, 1),
247 ragged_tensor.RaggedTensorSpec(None, dtypes.int32, 2)),
249 ragged_tensor.RaggedTensorSpec([3, None], dtypes.int32, 1),
250 ragged_tensor.RaggedTensorSpec([5, None], dtypes.int32, 1)),
252 ragged_tensor.RaggedTensorSpec(None, dtypes.int32, 1),
253 ragged_tensor.RaggedTensorSpec(None, dtypes.float32, 1)),
570 ragged_tensor.RaggedTensorSpec([2, None], dtypes.int32, 1),
571 ragged_tensor.RaggedTensorSpec([2, None], dtypes.int32, 1)),
633 ragged_tensor.RaggedTensorSpec([2, None], dtypes.float32, 1), 32,
[all …]
Dstructure.py64 return ragged_tensor.RaggedTensorSpec(shape, dtype, ragged_rank)
115 elif isinstance(spec, ragged_tensor.RaggedTensorSpec):
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-ragged-tensor-spec.pbtxt1 path: "tensorflow.RaggedTensorSpec"
3 is_instance: "<class \'tensorflow.python.ops.ragged.ragged_tensor.RaggedTensorSpec\'>"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-ragged-tensor-spec.pbtxt1 path: "tensorflow.RaggedTensorSpec"
3 is_instance: "<class \'tensorflow.python.ops.ragged.ragged_tensor.RaggedTensorSpec\'>"
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor_spec_test.py42 R_1_N = ragged_tensor.RaggedTensorSpec([1, None])
43 R_1_N_N = ragged_tensor.RaggedTensorSpec([1, None, None])
44 R_2_1_N = ragged_tensor.RaggedTensorSpec([2, 1, None])
Dstructured_tensor.py912 elif isinstance(typespec, ragged_tensor.RaggedTensorSpec):
1058 ragged_tensor.RaggedTensorSpec)):
Dstructured_tensor_test.py535 ragged_tensor.RaggedTensorSpec([2, 2, None],
580 "c": ragged_tensor.RaggedTensorSpec([None, None], dtypes.int32)}),
/external/tensorflow/tensorflow/python/ops/
Dmap_fn.py538 elif isinstance(spec, ragged_tensor.RaggedTensorSpec):
540 return ragged_tensor.RaggedTensorSpec(None, spec._dtype, spec._ragged_rank,
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dparsing_ops.py84 self._element_spec[key] = ragged_tensor.RaggedTensorSpec(
Dbatching.py413 return ragged_tensor.RaggedTensorSpec(
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dcategory_crossing.py193 isinstance(inp_spec, ragged_tensor.RaggedTensorSpec)
/external/tensorflow/tensorflow/python/saved_model/
Dnested_structure_coder_test.py197 structure = [ragged_tensor.RaggedTensorSpec(
326 {"rt": ragged_tensor.RaggedTensorSpec([10, None], dtypes.int32),
Dnested_structure_coder.py504 ragged_tensor.RaggedTensorSpec,
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_type_spec_test.py385 ragged_tensor_lib.RaggedTensorSpec(
388 ragged_tensor_lib.RaggedTensorSpec(
Dinput_lib.py597 spec, (sparse_tensor.SparseTensorSpec, ragged_tensor.RaggedTensorSpec)):
1470 if isinstance(spec, ragged_tensor.RaggedTensorSpec):
1487 if dims and (isinstance(spec, ragged_tensor.RaggedTensorSpec) or
1499 if isinstance(spec, ragged_tensor.RaggedTensorSpec):
/external/tensorflow/tensorflow/core/protobuf/
Dstruct.proto131 RAGGED_TENSOR_SPEC = 3; // tf.RaggedTensorSpec
/external/tensorflow/tensorflow/python/kernel_tests/
Dmap_fn_test.py91 fn_output_signature=ragged_tensor.RaggedTensorSpec([None], rt.dtype))
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dload.py1121 elif isinstance(x, ragged_tensor.RaggedTensorSpec):
1122 return ragged_tensor.RaggedTensorSpec(common_shape, x.dtype)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_generator_test.py464 output_signature=ragged_tensor.RaggedTensorSpec(
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py465 ragged_tensor.RaggedTensorSpec([2, None], dtypes.int32))
472 ragged_tensor.RaggedTensorSpec([3, None], dtypes.int32))
475 ragged_tensor.RaggedTensorSpec([None, None], dtypes.int32))
480 ragged_tensor.RaggedTensorSpec([1, None, None], dtypes.int32))
483 ragged_tensor.RaggedTensorSpec([None, None, None], dtypes.int32))
1519 [ragged_tensor.RaggedTensorSpec([None, None], dtypes.int32)]),
1523 [ragged_tensor.RaggedTensorSpec([None, None, None], dtypes.int32)]),
2378 signature = [ragged_tensor.RaggedTensorSpec(
5082 't': (ragged_tensor.RaggedTensorSpec([2, None, None], dtypes.int32),

12