Home
last modified time | relevance | path

Searched refs:type_spec (Results 1 – 25 of 94) sorted by relevance

1234

/external/tensorflow/tensorflow/python/keras/engine/
Dkeras_tensor.py27 from tensorflow.python.framework import type_spec as type_spec_module
143 def __init__(self, type_spec, inferred_value=None, name=None): argument
145 if not isinstance(type_spec, type_spec_module.TypeSpec):
148 self._type_spec = type_spec
153 def type_spec(self): member in KerasTensor
171 type_spec = type_spec_module.type_spec_from_value(tensor)
173 if (type_spec.dtype == dtypes.int32 and type_spec.shape.rank is not None
174 and type_spec.shape.rank < 2):
201 return KerasTensor(type_spec, inferred_value=inferred_value, name=name)
205 type_spec = type_spec_module.type_spec_from_value(tensor)
[all …]
Dinput_layer_test.py27 from tensorflow.python.framework import type_spec
76 @type_spec.register('tf.TwoTensorsSpec')
77 class TwoTensorsSpecNoOneDtype(type_spec.TypeSpec):
117 type_spec.register_type_spec_from_value_converter(
239 type_spec=tensor_spec.TensorSpec((2, 7, 32), dtypes.float32))
244 type_spec=tensor_spec.TensorSpec((7, 32), dtypes.float32))
249 type_spec=tensor_spec.TensorSpec((7, 32), dtypes.float32))
254 type_spec=tensor_spec.TensorSpec((7, 32), dtypes.float32))
259 type_spec=tensor_spec.TensorSpec((7, 32), dtypes.float32))
265 type_spec=tensor_spec.TensorSpec((7, 32), dtypes.float32))
[all …]
Dinput_layer.py109 type_spec=None, argument
116 self._init_type_spec = type_spec
166 if type_spec is not None:
180 input_tensor = keras_tensor.keras_tensor_from_type_spec(type_spec)
265 type_spec=None, argument
363 'type_spec': type_spec}
371 and type_spec is None):
Dkeras_tensor_test.py36 type_spec=tensor_spec.TensorSpec(shape=(1, 2, 3), dtype=dtypes.float32))
44 type_spec=tensor_spec.TensorSpec(shape=(2,), dtype=dtypes.int32),
54 type_spec=sparse_tensor.SparseTensorSpec(
/external/robolectric-shadows/resources/src/main/java/org/robolectric/res/android/
DLoadedArsc.java86 ResTable_typeSpec type_spec; field in LoadedArsc.TypeSpec
102 if (entry_index >= dtohl(type_spec.entryCount)) { in GetFlagsForEntryIndex()
107 int[] flags = type_spec.getSpecFlags(); in GetFlagsForEntryIndex()
200 TypeSpec type_spec = new TypeSpec(); in Build() local
201 type_spec.types = new ResTable_type[types_.size()]; in Build()
202 type_spec.type_spec = header_; in Build()
203 type_spec.idmap_entries = idmap_header_; in Build()
204 type_spec.type_count = types_.size(); in Build()
206 for (int i = 0; i < type_spec.types.length; i++) { in Build()
207 type_spec.types[i] = types_.get(i); in Build()
[all …]
/external/tensorflow/tensorflow/python/framework/
Dtype_spec_test.py34 from tensorflow.python.framework import type_spec
55 @type_spec.register("tf.TwoTensorsSpec")
56 class TwoTensorsSpec(type_spec.TypeSpec):
89 type_spec.register_type_spec_from_value_converter(
106 @type_spec.register("tf.TwoCompositesSpec")
107 class TwoCompositesSpec(type_spec.TypeSpec):
133 return cls(type_spec.type_spec_from_value(value.x),
134 type_spec.type_spec_from_value(value.y),
138 type_spec.register_type_spec_from_value_converter(
149 @type_spec.register("tf.NestOfTensorsSpec")
[all …]
Dtensor_spec.py27 from tensorflow.python.framework import type_spec
32 class DenseSpec(type_spec.TypeSpec):
117 @type_spec.register("tf.TensorSpec")
118 class TensorSpec(DenseSpec, type_spec.BatchableTypeSpec):
223 @type_spec.register("tf.BoundedTensorSpec")
348 type_spec.register_type_spec_from_value_converter(
352 type_spec.register_type_spec_from_value_converter(
Dsparse_tensor.py34 from tensorflow.python.framework import type_spec
295 @type_spec.register("tf.SparseTensorSpec")
296 class SparseTensorSpec(type_spec.BatchableTypeSpec):
434 type_spec.register_type_spec_from_value_converter(
436 type_spec.register_type_spec_from_value_converter(
/external/tensorflow/tensorflow/core/framework/
Dextension_type_variant.cc43 string type_spec; in DebugString() local
46 printer.PrintToString(metadata_.type_spec_proto(), &type_spec); in DebugString()
48 result.append(type_spec.empty() ? "none" : type_spec); in DebugString()
Dextension_type_variant_test.cc45 TypeSpecProto type_spec; in Make2DRaggedTensor() local
47 protobuf::TextFormat::ParseFromString(k2DRaggedTensorSpec, &type_spec)); in Make2DRaggedTensor()
51 ExtensionTypeVariant v(type_spec, absl::MakeSpan(components)); in Make2DRaggedTensor()
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Djson_utils.py36 from tensorflow.python.framework import type_spec
75 return type_spec.lookup(obj['type_spec'])._deserialize( # pylint: disable=protected-access
134 if isinstance(obj, type_spec.TypeSpec):
136 type_spec_name = type_spec.get_name(type(obj))
/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddistribute.py193 def get_batch_dim(type_spec): argument
194 shape = type_spec._to_legacy_output_shapes() # pylint: disable=protected-access
253 def recalculate_batch_size(type_spec): argument
255 output_shape = type_spec._to_legacy_output_shapes() # pylint: disable=protected-access
277 def rebatch(type_spec): argument
279 batch_size = recalculate_batch_size(type_spec)
280 return type_spec._unbatch()._batch(batch_size)
/external/tensorflow/tensorflow/python/ops/
Dmap_fn.py33 from tensorflow.python.framework import type_spec
385 elems_flat_signature = [type_spec.type_spec_from_value(e) for e in elems_flat]
528 if not isinstance(d, type_spec.TypeSpec):
553 if not isinstance(spec, type_spec.BatchableTypeSpec):
563 spec = type_spec.type_spec_from_value(elems_tensor)
564 if not isinstance(spec, type_spec.BatchableTypeSpec):
599 (r_spec, type_spec.type_spec_from_value(r_value), r_value))
/external/tensorflow/tensorflow/python/data/util/
Dstructure.py32 from tensorflow.python.framework import type_spec
166 if isinstance(flat_class, type_spec.TypeSpec):
437 spec = type_spec._type_spec_from_value(element) # pylint: disable=protected-access
495 @type_spec.register("tf.NoneTensorSpec")
496 class NoneTensorSpec(type_spec.BatchableTypeSpec):
548 type_spec.register_type_spec_from_value_converter(type(None),
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.-dataset-spec.pbtxt4 is_instance: "<class \'tensorflow.python.framework.type_spec.BatchableTypeSpec\'>"
5 is_instance: "<class \'tensorflow.python.framework.type_spec.TypeSpec\'>"
Dtensorflow.-sparse-tensor-spec.pbtxt4 is_instance: "<class \'tensorflow.python.framework.type_spec.BatchableTypeSpec\'>"
5 is_instance: "<class \'tensorflow.python.framework.type_spec.TypeSpec\'>"
Dtensorflow.-ragged-tensor-spec.pbtxt4 is_instance: "<class \'tensorflow.python.framework.type_spec.BatchableTypeSpec\'>"
5 is_instance: "<class \'tensorflow.python.framework.type_spec.TypeSpec\'>"
Dtensorflow.-tensor-spec.pbtxt5 is_instance: "<class \'tensorflow.python.framework.type_spec.BatchableTypeSpec\'>"
6 is_instance: "<class \'tensorflow.python.framework.type_spec.TypeSpec\'>"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.-dataset-spec.pbtxt4 is_instance: "<class \'tensorflow.python.framework.type_spec.BatchableTypeSpec\'>"
5 is_instance: "<class \'tensorflow.python.framework.type_spec.TypeSpec\'>"
Dtensorflow.data.experimental.-dataset-structure.pbtxt4 is_instance: "<class \'tensorflow.python.framework.type_spec.BatchableTypeSpec\'>"
5 is_instance: "<class \'tensorflow.python.framework.type_spec.TypeSpec\'>"
Dtensorflow.-sparse-tensor-spec.pbtxt4 is_instance: "<class \'tensorflow.python.framework.type_spec.BatchableTypeSpec\'>"
5 is_instance: "<class \'tensorflow.python.framework.type_spec.TypeSpec\'>"
Dtensorflow.-ragged-tensor-spec.pbtxt4 is_instance: "<class \'tensorflow.python.framework.type_spec.BatchableTypeSpec\'>"
5 is_instance: "<class \'tensorflow.python.framework.type_spec.TypeSpec\'>"
Dtensorflow.-tensor-spec.pbtxt5 is_instance: "<class \'tensorflow.python.framework.type_spec.BatchableTypeSpec\'>"
6 is_instance: "<class \'tensorflow.python.framework.type_spec.TypeSpec\'>"
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor_spec_test.py26 from tensorflow.python.framework import type_spec
242 unbatched_spec = type_spec.type_spec_from_value(batched)._unbatch()
244 unbatched_spec = type_spec.type_spec_from_value(unbatched[0])
256 batched_spec = type_spec.type_spec_from_value(batched)
Dstructured_tensor_test.py681 def testPyvalConversion(self, pyval, expected, type_spec=None): argument
683 actual = structured_tensor.StructuredTensor.from_pyval(pyval, type_spec)
744 type_spec=structured_tensor.StructuredTensorSpec(
750 type_spec=structured_tensor.StructuredTensorSpec(
756 type_spec=structured_tensor.StructuredTensorSpec(
762 type_spec=structured_tensor.StructuredTensorSpec(
767 type_spec=structured_tensor.StructuredTensorSpec(
773 type_spec=sparse_tensor.SparseTensorSpec([None], dtypes.int32),
777 type_spec=structured_tensor.StructuredTensorSpec(
789 def testFromPyvalError(self, pyval, err=ValueError, type_spec=None, msg=None): argument
[all …]

1234