| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 51 def empty_tensor_list(element_shape, argument
59 element_shape=_build_element_shape(element_shape),
65 def _set_handle_data(list_handle, element_shape, element_dtype): argument
70 if tensor_util.is_tf_type(element_shape):
71 element_shape = tensor_shape.TensorShape(None)
72 elif not isinstance(element_shape, tensor_shape.TensorShape):
73 element_shape = tensor_shape.TensorShape(element_shape)
78 shape=element_shape.as_proto(),
84 def tensor_list_reserve(element_shape, num_elements, element_dtype, name=None): argument
86 element_shape=_build_element_shape(element_shape),
[all …]
|
| D | tensor_array_ops.py | 65 element_shape=None, argument
112 if handle is not None and element_shape is not None:
140 self._element_shape = [tensor_shape.as_shape(element_shape)]
158 element_shape=element_shape,
183 def element_shape(self): member in _GraphTensorArray
196 if not shape.is_compatible_with(self.element_shape):
198 (shape, self.element_shape))
200 self._element_shape[0] = self.element_shape.merge_with(shape)
287 if (self.element_shape and not self._dynamic_size and
290 self.element_shape.dims)
[all …]
|
| D | tensor_array_ops_test.py | 37 size=4, dtype=dtypes.string, element_shape=[None], infer_shape=False)
55 size=4, dtype=dtypes.string, element_shape=[None], infer_shape=False)
68 size=4, dtype=dtypes.string, element_shape=[None], infer_shape=False)
84 element_shape=np.array((2, 3)))
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | data_structures.py | 57 def tf_tensor_array_new(elements, element_dtype=None, element_shape=None): argument
79 if element_shape is not None and element_shape != inferred_shape:
82 element_shape, elements, inferred_shape))
93 if element_shape is None:
94 element_shape = inferred_shape
100 infer_shape=(element_shape is None),
101 element_shape=element_shape)
107 def tf_tensor_list_new(elements, element_dtype=None, element_shape=None): argument
110 if element_shape is not None:
113 element_shape = array_ops.shape(elements)[1:]
[all …]
|
| D | data_structures_test.py | 52 element_shape=())
73 data_structures.tf_tensor_list_new([3, 4], element_shape=(2,))
76 constant_op.constant([1, 2, 3]), element_shape=[1])
92 data_structures.tf_tensor_array_new([3, 4], element_shape=(2,))
94 data_structures.tf_tensor_array_new([], element_shape=(2,))
127 l = list_ops.tensor_list_from_tensor(initial_list, element_shape=elem_shape)
131 element_shape=(2,))
145 opts = data_structures.ListPopOpts(element_dtype=None, element_shape=())
152 l = list_ops.tensor_list_from_tensor(initial_list, element_shape=elem_shape)
164 element_shape=None, element_dtype=dtypes.variant)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 67 shape_inference::ShapeHandle element_shape; in __anonabedecb60202() local
69 0, &element_shape)); in __anonabedecb60202()
72 {element_shape, element_dtype, ST_TENSOR_LIST}}); in __anonabedecb60202()
85 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anonabedecb60302() local
104 c->Merge(element_shape, list_shape_type.shape, &ignored)); in __anonabedecb60302()
105 element_shape = list_shape_type.shape; in __anonabedecb60302()
109 {element_shape, element_dtype, ST_TENSOR_LIST}}); in __anonabedecb60302()
132 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anonabedecb60402() local
151 c->Merge(element_shape, list_shape_type.shape, &ignored)); in __anonabedecb60402()
152 element_shape = list_shape_type.shape; in __anonabedecb60402()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | list_ops_test.py | 57 element_shape=[],
81 element_dtype=dtypes.float32, element_shape=[], max_num_elements=1)
94 element_shape=[],
103 element_dtype=dtypes.float32, element_shape=[2, 3], num_elements=3)
112 element_dtype=dtypes.float32, element_shape=[None, 3], num_elements=3)
114 l, element_dtype=dtypes.float32, element_shape=[4, 3])
119 element_dtype=dtypes.float32, element_shape=None, num_elements=3)
128 element_dtype=dtypes.float32, element_shape=[None, 2], num_elements=3)
133 l, element_dtype=dtypes.float32, element_shape=[1, 3])
139 element_dtype=dtypes.float32, element_shape=None)
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_list.h | 71 : element_shape(other.element_shape), in TensorList()
79 : element_shape(std::move(rhs.element_shape)), in TensorList()
88 element_shape = rhs.element_shape;
99 element_shape = rhs.element_shape;
117 PartialTensorShape element_shape; variable
132 out.element_shape = element_shape; in Copy()
|
| D | list_kernels.h | 49 PartialTensorShape* element_shape);
96 if (!tensor_list->element_shape.IsFullyDefined()) { in Compute()
108 TensorShape element_shape; in Compute() local
109 OP_REQUIRES(c, partial_element_shape.AsTensorShape(&element_shape), in Compute()
114 TensorShape output_shape = element_shape; in Compute()
136 c, c->allocate_temp(element_dtype_, element_shape, &zeros, attr)); in Compute()
187 TensorShape element_shape; in Compute() local
206 c, partial_element_shape.AsTensorShape(&element_shape), in Compute()
216 OP_REQUIRES_OK(c, c->allocate_output(0, element_shape, &result, attr)); in Compute()
252 TensorShape element_shape; in Compute() local
[all …]
|
| D | list_kernels.cc | 67 PartialTensorShape* element_shape) { in GetElementShapeFromInput() argument
68 TF_RETURN_IF_ERROR(TensorShapeFromTensor(c->input(index), element_shape)); in GetElementShapeFromInput()
71 PartialTensorShape tmp = *element_shape; in GetElementShapeFromInput()
72 TF_RETURN_IF_ERROR(tmp.MergeWith(tensor_list.element_shape, element_shape)); in GetElementShapeFromInput()
149 PartialTensorShape element_shape; in Compute() local
150 OP_REQUIRES_OK(ctx, TensorShapeFromTensor(ctx->input(0), &element_shape)); in Compute()
151 empty.element_shape = element_shape; in Compute()
190 OP_REQUIRES(c, l->element_shape.IsCompatibleWith(input.shape()), in Compute()
195 " list shape: ", l->element_shape.DebugString())); in Compute()
262 if (l->element_shape.unknown_rank()) { in Compute()
[all …]
|
| D | deserialize_sparse_variant_op.cc | 102 const Tensor* element_shape; in Compute() local
105 &element_shape, &total_non_zeros)); in Compute()
108 2, {input_dims_to_stack + element_shape->NumElements()}, in Compute()
110 const auto element_shape_vec = element_shape->vec<int64>(); in Compute()
116 for (int64 j = 0; j < element_shape->NumElements(); ++j) { in Compute()
132 i, &element_shape, &num_non_zeros)); in Compute()
137 element_shape->NumElements(), in Compute()
143 "] is: ", element_shape->NumElements())); in Compute()
144 const auto element_shape_vec = element_shape->vec<int64>(); in Compute()
145 for (int j = 0; j < element_shape->NumElements(); ++j) { in Compute()
|
| D | tensor_list.cc | 49 element_shape.AsProto(&element_shape_proto); in Encode()
57 to->element_shape = from.element_shape; in TensorListDeviceCopy()
121 element_shape = PartialTensorShape(element_shape_proto); in Decode()
|
| D | tensor_array_ops.cc | 332 auto element_shape = PartialTensorShape(); in CreateTensorArray() local
345 gtl::ArraySlice<int64>(dims), &element_shape)); in CreateTensorArray()
348 element_shape = tensor_array->ElemShape(); in CreateTensorArray()
352 auto creator = [key, tensor_array, array_size, marked_size, element_shape, in CreateTensorArray()
357 array_size, element_shape, tensor_array->HasIdenticalElementShapes(), in CreateTensorArray()
1039 TensorShape element_shape(tensor_value->shape()); in Compute() local
1042 FastBoundsCheck(element_shape.dim_size(0), in Compute()
1052 OP_REQUIRES(ctx, element_shape.dims() > 0, in Compute()
1055 element_shape.DebugString())); in Compute()
1063 num_values = element_shape.dim_size(0); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/python/autograph/lang/ |
| D | special_functions.py | 31 def _validate_list_constructor(elements, element_dtype, element_shape): argument
33 if element_dtype is not None and element_shape is not None:
59 element_shape=None, argument
83 _validate_list_constructor(elements, element_dtype, element_shape)
86 element_shape)
89 element_shape)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_list_ops_test.py | 40 element_shape=(dim, 15),
51 element_shape=(7, 15),
67 element_shape=(7, 15),
85 element_shape=[],
102 element_shape=(7, 15), element_dtype=dtypes.float32)
113 element_shape=(10, 15), element_dtype=dtypes.float32,
123 l = list_ops.tensor_list_from_tensor(t, element_shape=[])
135 l = list_ops.tensor_list_from_tensor(t, element_shape=[])
145 element_shape=[], element_dtype=dtypes.float32, max_num_elements=2)
156 element_dtype=dtypes.float32, element_shape=[], num_elements=2)
[all …]
|
| D | add_n_test.py | 35 element_shape=[], element_dtype=dtypes.float32, num_elements=3)
37 element_shape=[], element_dtype=dtypes.float32, num_elements=3)
49 element_shape=[], element_dtype=dtypes.float32, num_elements=2)
51 element_shape=[], element_dtype=dtypes.float32, num_elements=3)
65 element_shape=l1_element_shape,
69 element_shape=l2_element_shape,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_utils.cc | 247 Status GetTensorListShapeFromElementShape(const xla::Shape& element_shape, in GetTensorListShapeFromElementShape() argument
251 if (!element_shape.IsArray()) { in GetTensorListShapeFromElementShape()
255 element_shape.DebugString()); in GetTensorListShapeFromElementShape()
258 std::vector<int64> dimensions = xla::SpanToVector(element_shape.dimensions()); in GetTensorListShapeFromElementShape()
261 xla::ShapeUtil::MakeShape(element_shape.element_type(), dimensions)); in GetTensorListShapeFromElementShape()
307 TF_ASSIGN_OR_RETURN(xla::Shape element_shape, b->GetShape(element)); in GetInitializedTensorListForElement()
311 element_shape, leading_dim, leading_dim_is_dynamic, &list_shape)); in GetInitializedTensorListForElement()
314 element_shape, leading_dim, leading_dim_is_dynamic, &list_shape)); in GetInitializedTensorListForElement()
369 TF_ASSIGN_OR_RETURN(xla::Shape element_shape, b->GetShape(element)); in ExecuteTensorListPushBack()
370 int element_tuple_size = xla::ShapeUtil::TupleElementCount(element_shape); in ExecuteTensorListPushBack()
[all …]
|
| D | tensor_list_ops.cc | 52 XlaOpKernelContext* ctx, const xla::Shape& element_shape, in GetTensorListDynamicDims() argument
72 for (int64 dim = 0; dim < element_shape.dimensions_size(); ++dim) { in GetTensorListDynamicDims()
166 xla::Shape element_shape; in Compile() local
169 &got_shape, &element_shape)); in Compile()
173 element_shape, num_elements, in Compile()
177 ctx, element_shape, list_shape, num_elements); in Compile()
238 xla::Shape element_shape; in Compile() local
241 &got_shape, &element_shape)); in Compile()
245 element_shape, max_num_elements, in Compile()
249 ctx, element_shape, list_shape, max_num_elements); in Compile()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | tensor-list.pbtxt | 82 name: "TensorListStack/element_shape"
105 name: "TensorListReserve/element_shape"
133 input: "TensorListReserve/element_shape"
153 input: "TensorListReserve/element_shape"
189 input: "TensorListStack/element_shape"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TensorListConcatV2.pbtxt | 8 element_shape: The shape of the uninitialized elements in the list. If the first
12 the leading dim of input_handle.element_shape or the element_shape input arg
|
| D | api_def_EmptyTensorList.pbtxt | 6 with element_shape.
10 element_shape: a shape compatible with that of elements in the list.
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure_test.py | 61 dtype=dtypes.float32, element_shape=(3,), size=0),
109 dtype=dtypes.float32, element_shape=(3,), size=0), lambda: [
111 dtype=dtypes.float32, element_shape=(3,), size=0),
113 dtype=dtypes.float32, element_shape=(3,), size=10)
116 dtype=dtypes.int32, element_shape=(3,), size=0),
118 dtype=dtypes.float32, element_shape=(), size=0)
192 dtype=dtypes.float32, element_shape=(3,), size=0),
194 dtype=dtypes.float32, element_shape=(3,), size=0),
196 dtype=dtypes.int32, element_shape=(), size=0)),
264 dtype=dtypes.float32, element_shape=(3,), size=0),
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion_test.cc | 1902 const Shape element_shape = ShapeUtil::MakeShape(F32, {42}); in TEST_F() local
1904 {element_shape, element_shape, element_shape, element_shape}); in TEST_F()
1909 HloInstruction::CreateParameter(0, element_shape, "param_0")); in TEST_F()
1911 HloInstruction::CreateParameter(1, element_shape, "param_1")); in TEST_F()
1913 HloInstruction::CreateParameter(2, element_shape, "param_2")); in TEST_F()
1915 HloInstruction::CreateParameter(3, element_shape, "param_3")); in TEST_F()
1931 HloInstruction::CreateGetTupleElement(element_shape, body_param, 0)); in TEST_F()
1933 HloInstruction::CreateGetTupleElement(element_shape, body_param, 1)); in TEST_F()
1935 HloInstruction::CreateGetTupleElement(element_shape, body_param, 2)); in TEST_F()
1937 HloInstruction::CreateGetTupleElement(element_shape, body_param, 3)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TensorArrayGatherV2.pbtxt | 24 name: "element_shape"
56 name: "element_shape"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TensorArrayGatherV2.pbtxt | 24 name: "element_shape"
56 name: "element_shape"
|