Home
last modified time | relevance | path

Searched refs:StructuredTensorSpec (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor_spec_test.py32 from tensorflow.python.ops.structured.structured_tensor import StructuredTensorSpec
75 spec1 = StructuredTensorSpec([1, 2, 3], spec1_fields)
80 spec2 = StructuredTensorSpec([1, 2], spec2_fields)
94 structured_tensor.StructuredTensorSpec(shape, field_specs)
97 spec1 = StructuredTensorSpec([1, 2, 3], dict(a=T_1_2))
101 (StructuredTensorSpec([1, 2, 3], {}),
103 (StructuredTensorSpec([], {'a': T_1_2}),
105 (StructuredTensorSpec([1, 2], {'a': T_1_2, 'b': R_1_N}),
107 (StructuredTensorSpec([], {'a': T_1_2}),
118 (StructuredTensorSpec([1, 2, 3], {}), {}),
[all …]
Dstructured_tensor_slice_test.py112 EXAMPLE_STRUCT_SPEC1 = structured_tensor.StructuredTensorSpec([], {
115 "f3": structured_tensor.StructuredTensorSpec(
117 "f4": structured_tensor.StructuredTensorSpec(
120 "f5": structured_tensor.StructuredTensorSpec(
Dstructured_tensor_test.py530 structured_tensor.StructuredTensorSpec(
577 "type_spec": structured_tensor.StructuredTensorSpec([], {
609 "type_spec": structured_tensor.StructuredTensorSpec([0], {
617 "type_spec": structured_tensor.StructuredTensorSpec([0, 5], {}),
624 "type_spec": structured_tensor.StructuredTensorSpec([1, 0], {}),
675 "type_spec": structured_tensor.StructuredTensorSpec([2, 3], {
744 type_spec=structured_tensor.StructuredTensorSpec(
750 type_spec=structured_tensor.StructuredTensorSpec(
756 type_spec=structured_tensor.StructuredTensorSpec(
762 type_spec=structured_tensor.StructuredTensorSpec(
[all …]
Dstructured_tensor.py868 if not (isinstance(typespec, StructuredTensorSpec) and
888 if ((not isinstance(typespec, StructuredTensorSpec)) or
920 elif isinstance(typespec, StructuredTensorSpec):
1027 return StructuredTensorSpec.from_value(self)
1030 class StructuredTensorSpec(type_spec.BatchableTypeSpec): class
1057 if not isinstance(value, (StructuredTensorSpec, tensor_spec.TensorSpec,
1090 return StructuredTensorSpec(
1096 return StructuredTensorSpec(
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py502 structured_tensor.StructuredTensorSpec(
506 structured_tensor.StructuredTensorSpec(
510 structured_tensor.StructuredTensorSpec(
516 structured_tensor.StructuredTensorSpec(
520 structured_tensor.StructuredTensorSpec(