Searched refs:field_specs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 867 field_specs = typespec._field_specs # pylint: disable=protected-access
869 spec_shape.rank == 0 and set(pyval) == set(field_specs)):
873 (k, cls.from_pyval(v, field_specs[k])) for (k, v) in pyval.items())
887 field_specs = typespec._field_specs # pylint: disable=protected-access
889 (set(fields) - set(field_specs))):
896 for (key, spec) in field_specs.items():
1035 def __init__(self, shape, field_specs): argument
1052 if not isinstance(field_specs, dict):
1054 for key, value in field_specs.items():
1063 self._field_specs = dict(field_specs)
[all …]
|
| D | structured_tensor_spec_test.py | 92 def testConstructionErrors(self, shape, field_specs, error): argument
94 structured_tensor.StructuredTensorSpec(shape, field_specs)
148 def testToFromComponents(self, shape, fields, field_specs): argument
151 spec = StructuredTensorSpec(shape, field_specs)
|
| D | structured_tensor_test.py | 746 field_specs={"b": tensor_spec.TensorSpec([], dtypes.int32)}),
752 field_specs={"b": tensor_spec.TensorSpec([], dtypes.int32)}),
758 field_specs={"a": tensor_spec.TensorSpec([], dtypes.int32)}),
763 shape=[], field_specs={}),
769 field_specs={"a": tensor_spec.TensorSpec([], dtypes.int32)}),
779 field_specs={"a": tensor_spec.TensorSpec([], dtypes.int32)}),
|