Home
last modified time | relevance | path

Searched refs:StructuredTensor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor_test.py44 from tensorflow.python.ops.structured.structured_tensor import StructuredTensor
54 if not (isinstance(a, structured_tensor.StructuredTensor) or
55 isinstance(b, structured_tensor.StructuredTensor)):
57 if not isinstance(a, structured_tensor.StructuredTensor):
58 a = structured_tensor.StructuredTensor.from_pyval(a)
60 elif not isinstance(b, structured_tensor.StructuredTensor):
61 b = structured_tensor.StructuredTensor.from_pyval(b)
74 if isinstance(a_value, structured_tensor.StructuredTensor):
82 structured_tensor.StructuredTensor({}, (), None, ())
113 "foo": StructuredTensor.from_pyval({"a": 1, "b": [1, 2, 3]}),
[all …]
Dstructured_tensor_spec_test.py31 from tensorflow.python.ops.structured.structured_tensor import StructuredTensor
55 if not (isinstance(a, structured_tensor.StructuredTensor) or
56 isinstance(b, structured_tensor.StructuredTensor)):
58 if not (isinstance(a, structured_tensor.StructuredTensor) and
59 isinstance(b, structured_tensor.StructuredTensor)):
98 self.assertEqual(spec1.value_type, StructuredTensor)
150 struct = StructuredTensor.from_fields(fields, shape)
181 StructuredTensor.from_fields({'a': 1, 'b': [5, 6]}),
182 StructuredTensor.from_fields({'a': 2, 'b': [7, 8]})],
184 'batched': lambda: StructuredTensor.from_fields(shape=[2], fields={
[all …]
Dstructured_tensor_slice_test.py130 if not (isinstance(a, structured_tensor.StructuredTensor) or
131 isinstance(b, structured_tensor.StructuredTensor)):
133 elif (isinstance(a, structured_tensor.StructuredTensor) and
134 isinstance(b, structured_tensor.StructuredTensor)):
141 elif isinstance(b, structured_tensor.StructuredTensor):
211 struct = structured_tensor.StructuredTensor.from_pyval(EXAMPLE_STRUCT)
215 struct2 = structured_tensor.StructuredTensor.from_pyval(
238 struct_vector = structured_tensor.StructuredTensor.from_pyval(
243 struct_vector2 = structured_tensor.StructuredTensor.from_pyval(
258 struct = structured_tensor.StructuredTensor.from_pyval(EXAMPLE_STRUCT)
[all …]
Dstructured_tensor.py45 class StructuredTensor(composite_tensor.CompositeTensor): class
415 if not isinstance(current_value, StructuredTensor):
435 return StructuredTensor.from_fields(
654 if not isinstance(value, StructuredTensor):
715 return StructuredTensor.from_fields(fields, self._shape)
742 return StructuredTensor.from_fields(new_fields, result_shape)
808 elif isinstance(value, StructuredTensor):
874 return StructuredTensor.from_fields(fields=fields, shape=(), validate=False)
898 return StructuredTensor.from_fields(
1067 return StructuredTensor
[all …]
Dstructured_array_ops.py25 from tensorflow.python.ops.structured.structured_tensor import StructuredTensor
30 @dispatch.dispatch_for_types(array_ops.expand_dims, StructuredTensor)
64 @dispatch.dispatch_for_types(array_ops.expand_dims_v2, StructuredTensor)
132 return StructuredTensor.from_fields(
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py501 structured_tensor.StructuredTensor.from_pyval({'a': [1]}),
505 structured_tensor.StructuredTensor.from_pyval({'b': [1]}),
509 structured_tensor.StructuredTensor.from_pyval({'c': [1]}),
515 structured_tensor.StructuredTensor.from_pyval({'a': [1, 2]}),
519 structured_tensor.StructuredTensor.from_pyval({'a': [1, 2, 3]}),
523 structured_tensor.StructuredTensor.from_pyval({'a': [1, 2, 3, 4]}),
/external/tensorflow/tensorflow/
DBUILD720 # TODO(b/159007891) Remove this package once StructuredTensor is exported.