/external/tensorflow/tensorflow/python/ops/ragged/ |
D | ragged_tensor_supported_values_test.py | 41 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 …]
|
D | ragged_tensor_test.py | 49 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 …]
|
D | ragged_map_ops.py | 179 return ragged_tensor.RaggedTensorSpec(
|
D | ragged_tensor.py | 2114 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)
|
D | ragged_print_op_test.py | 184 input_signature=[ragged_tensor.RaggedTensorSpec(ragged_rank=1)])
|
/external/tensorflow/tensorflow/python/data/util/ |
D | structure_test.py | 67 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 …]
|
D | structure.py | 64 return ragged_tensor.RaggedTensorSpec(shape, dtype, ragged_rank) 115 elif isinstance(spec, ragged_tensor.RaggedTensorSpec):
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.-ragged-tensor-spec.pbtxt | 1 path: "tensorflow.RaggedTensorSpec" 3 is_instance: "<class \'tensorflow.python.ops.ragged.ragged_tensor.RaggedTensorSpec\'>"
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.-ragged-tensor-spec.pbtxt | 1 path: "tensorflow.RaggedTensorSpec" 3 is_instance: "<class \'tensorflow.python.ops.ragged.ragged_tensor.RaggedTensorSpec\'>"
|
/external/tensorflow/tensorflow/python/ops/structured/ |
D | structured_tensor_spec_test.py | 42 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])
|
D | structured_tensor.py | 912 elif isinstance(typespec, ragged_tensor.RaggedTensorSpec): 1058 ragged_tensor.RaggedTensorSpec)):
|
D | structured_tensor_test.py | 535 ragged_tensor.RaggedTensorSpec([2, 2, None], 580 "c": ragged_tensor.RaggedTensorSpec([None, None], dtypes.int32)}),
|
/external/tensorflow/tensorflow/python/ops/ |
D | map_fn.py | 538 elif isinstance(spec, ragged_tensor.RaggedTensorSpec): 540 return ragged_tensor.RaggedTensorSpec(None, spec._dtype, spec._ragged_rank,
|
/external/tensorflow/tensorflow/python/data/experimental/ops/ |
D | parsing_ops.py | 84 self._element_spec[key] = ragged_tensor.RaggedTensorSpec(
|
D | batching.py | 413 return ragged_tensor.RaggedTensorSpec(
|
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
D | category_crossing.py | 193 isinstance(inp_spec, ragged_tensor.RaggedTensorSpec)
|
/external/tensorflow/tensorflow/python/saved_model/ |
D | nested_structure_coder_test.py | 197 structure = [ragged_tensor.RaggedTensorSpec( 326 {"rt": ragged_tensor.RaggedTensorSpec([10, None], dtypes.int32),
|
D | nested_structure_coder.py | 504 ragged_tensor.RaggedTensorSpec,
|
/external/tensorflow/tensorflow/python/distribute/ |
D | input_lib_type_spec_test.py | 385 ragged_tensor_lib.RaggedTensorSpec( 388 ragged_tensor_lib.RaggedTensorSpec(
|
D | input_lib.py | 597 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/ |
D | struct.proto | 131 RAGGED_TENSOR_SPEC = 3; // tf.RaggedTensorSpec
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | map_fn_test.py | 91 fn_output_signature=ragged_tensor.RaggedTensorSpec([None], rt.dtype))
|
/external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
D | load.py | 1121 elif isinstance(x, ragged_tensor.RaggedTensorSpec): 1122 return ragged_tensor.RaggedTensorSpec(common_shape, x.dtype)
|
/external/tensorflow/tensorflow/python/data/kernel_tests/ |
D | from_generator_test.py | 464 output_signature=ragged_tensor.RaggedTensorSpec(
|
/external/tensorflow/tensorflow/python/eager/ |
D | function_test.py | 465 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),
|