Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/structured/
H A Dstructured_tensor_test.py42 from tensorflow.python.ops.structured.structured_tensor import StructuredTensor
82 if not (isinstance(a, structured_tensor.StructuredTensor) or
83 isinstance(b, structured_tensor.StructuredTensor)):
85 if not isinstance(a, structured_tensor.StructuredTensor):
86 a = structured_tensor.StructuredTensor.from_pyval(a)
88 elif not isinstance(b, structured_tensor.StructuredTensor):
89 b = structured_tensor.StructuredTensor.from_pyval(b)
102 if isinstance(a_value, structured_tensor.StructuredTensor):
132 "foo": StructuredTensor.from_pyval({"a": 1, "b": [1, 2, 3]}),
133 "bar": StructuredTensor.from_pyval(
[all …]
H A Dstructured_tensor_spec_test.py33 from tensorflow.python.ops.structured.structured_tensor import StructuredTensor
61 if not (isinstance(a, structured_tensor.StructuredTensor) or
62 isinstance(b, structured_tensor.StructuredTensor)):
64 if not (isinstance(a, structured_tensor.StructuredTensor) and
65 isinstance(b, structured_tensor.StructuredTensor)):
82 spec1 = StructuredTensor.Spec(
92 spec2 = StructuredTensor.Spec(
111 structured_tensor.StructuredTensor.Spec(
119 spec1 = StructuredTensor.Spec(
125 self.assertEqual(spec1.value_type, StructuredTensor)
[all …]
H A Dstructured_array_ops.py28 from tensorflow.python.ops.structured.structured_tensor import StructuredTensor
34 def shape_v2(input: StructuredTensor, out_type=dtypes.int32, # pylint: disable=redefined-builtin argument
42 def shape_v1(input: StructuredTensor, name=None, # pylint: disable=redefined-builtin argument
49 @dispatch.dispatch_for_types(array_ops.expand_dims, StructuredTensor)
83 @dispatch.dispatch_for_types(array_ops.expand_dims_v2, StructuredTensor)
114 @dispatch.dispatch_for_types(array_ops.gather, StructuredTensor)
159 @dispatch.dispatch_for_types(array_ops.concat, StructuredTensor)
184 @dispatch.dispatch_for_types(random_ops.random_shuffle, StructuredTensor)
205 @dispatch.dispatch_for_types(array_ops.size_v2, StructuredTensor)
213 @dispatch.dispatch_for_types(array_ops.size, StructuredTensor)
[all …]
H A Dstructured_array_ops_test.py34 from tensorflow.python.ops.structured.structured_tensor import StructuredTensor
46 if not (isinstance(a, structured_tensor.StructuredTensor) or
47 isinstance(b, structured_tensor.StructuredTensor)):
50 if not isinstance(a, structured_tensor.StructuredTensor):
51 a = structured_tensor.StructuredTensor.from_pyval(a)
52 elif not isinstance(b, structured_tensor.StructuredTensor):
53 b = structured_tensor.StructuredTensor.from_pyval(b)
76 if isinstance(a_value, structured_tensor.StructuredTensor):
161 st = StructuredTensor.from_pyval(st)
167 st = StructuredTensor.from_pyval(st)
[all …]
H A Dstructured_tensor.py41 class StructuredTensor(extension_type.BatchableExtensionType): class
138 return StructuredTensor(
155 return StructuredTensor(fields={}, ragged_shape=ragged_shape)
298 return StructuredTensor(fields=fields, ragged_shape=shape)
444 if not isinstance(current_value, StructuredTensor):
464 return StructuredTensor.from_fields(
688 if not isinstance(value, StructuredTensor):
750 return StructuredTensor.from_fields(fields, self.shape)
777 return StructuredTensor.from_fields(new_fields, result_shape)
842 elif isinstance(value, StructuredTensor):
[all …]
H A Dstructured_tensor_slice_test.py126 if not (isinstance(a, structured_tensor.StructuredTensor) or
127 isinstance(b, structured_tensor.StructuredTensor)):
129 elif (isinstance(a, structured_tensor.StructuredTensor) and
130 isinstance(b, structured_tensor.StructuredTensor)):
137 elif isinstance(b, structured_tensor.StructuredTensor):
207 struct = structured_tensor.StructuredTensor.from_pyval(EXAMPLE_STRUCT)
211 struct2 = structured_tensor.StructuredTensor.from_pyval(
234 struct_vector = structured_tensor.StructuredTensor.from_pyval(
239 struct_vector2 = structured_tensor.StructuredTensor.from_pyval(
254 struct = structured_tensor.StructuredTensor.from_pyval(EXAMPLE_STRUCT)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dtype_utils.py23 from tensorflow.python.ops.structured.structured_tensor import StructuredTensor
117 if isinstance(element_spec, StructuredTensor.Spec):
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dfunction_test.py517 structured_tensor.StructuredTensor.from_pyval({'a': [1]}),
518 structured_tensor.StructuredTensor.Spec._from_fields_and_rank(
522 structured_tensor.StructuredTensor.from_pyval({'b': [1]}),
523 structured_tensor.StructuredTensor.Spec._from_fields_and_rank(
527 structured_tensor.StructuredTensor.from_pyval({'c': [1]}),
528 structured_tensor.StructuredTensor.Spec._from_fields_and_rank(
534 structured_tensor.StructuredTensor.from_pyval({'a': [1, 2]}),
535 structured_tensor.StructuredTensor.Spec._from_fields_and_rank(
539 structured_tensor.StructuredTensor.from_pyval({'a': [1, 2, 3]}),
542 structured_tensor.StructuredTensor.from_pyval({'a': [1, 2, 3, 4]}),
/aosp_15_r20/external/tensorflow/tensorflow/
H A DBUILD963 # TODO(b/159007891) Remove this package once StructuredTensor is exported.