| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 61 shape_inference::ShapeHandle element_shape; in __anon9187032f0202() local
79 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anon9187032f0302() local
126 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anon9187032f0402() local
205 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anon9187032f0602() local
242 shape_inference::ShapeHandle element_shape) { in TensorListConcatShapeInference()
285 shape_inference::ShapeHandle element_shape; in __anon9187032f0702() local
300 shape_inference::ShapeHandle element_shape; in __anon9187032f0802() local
329 shape_inference::ShapeHandle element_shape; in __anon9187032f0902() local
355 shape_inference::ShapeHandle element_shape; in __anon9187032f0a02() local
388 shape_inference::ShapeHandle element_shape; in __anon9187032f0c02() local
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 45 def empty_tensor_list(element_shape, argument
59 def tensor_list_reserve(element_shape, num_elements, element_dtype, name=None): argument
67 def tensor_list_from_tensor(tensor, element_shape, name=None): argument
74 def tensor_list_get_item(input_handle, index, element_dtype, element_shape=None, argument
95 element_shape=None, argument
107 element_shape=None, argument
125 element_shape=None, argument
135 def tensor_list_concat(input_handle, element_dtype, element_shape=None, argument
147 def tensor_list_split(tensor, element_shape, lengths, name=None): argument
|
| D | tensor_array_ops.py | 58 element_shape=None, argument
427 element_shape=None, argument
689 element_shape=None, argument
976 element_shape=None, argument
1074 def _element_shape(self, element_shape): argument
|
| D | rnn.py | 779 def _create_ta(name, element_shape, dtype): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 60 PartialTensorShape element_shape; member
129 TensorShape element_shape; in Compute() local
208 TensorShape element_shape; in Compute() local
273 TensorShape element_shape; in Compute() local
309 PartialTensorShape element_shape; in TensorListConcat() local
334 PartialTensorShape element_shape; in Compute() local
466 TensorShape element_shape = output_shape; in Compute() local
505 PartialTensorShape element_shape; in Compute() local
612 TensorShape element_shape; in Compute() local
680 PartialTensorShape element_shape; in Compute() local
[all …]
|
| D | list_kernels.cc | 170 PartialTensorShape* element_shape) { in GetElementShapeFromInput()
239 PartialTensorShape element_shape; in Compute() local
385 PartialTensorShape element_shape; in Compute() local
|
| D | tensor_array.h | 141 int32 N, const PartialTensorShape& element_shape, in TensorArray()
562 TensorShape element_shape; in LockedRead() local
|
| D | deserialize_sparse_variant_op.cc | 102 const Tensor* element_shape; in Compute() local
|
| D | fifo_queue.cc | 92 TensorShape element_shape(tuple[component].shape()); in GetElementComponentFromBatch() local
|
| D | barrier_ops.cc | 92 TensorShape element_shape = values.shape(); in TryInsertMany() local
307 const Tensor& values, const TensorShape& element_shape, in InsertOneLocked()
|
| D | priority_queue.cc | 120 TensorShape element_shape(tuple[component].shape()); in GetElementComponentFromBatch() local
|
| D | padding_fifo_queue.cc | 61 TensorShape element_shape(tuple[component].shape()); in GetElementComponent() local
|
| D | random_shuffle_queue_op.cc | 168 TensorShape element_shape(tuple[component].shape()); in GetElementComponentFromBatch() local
|
| D | tensor_array_ops.cc | 334 auto element_shape = PartialTensorShape(); in CreateTensorArray() local
1046 TensorShape element_shape(tensor_value->shape()); in Compute() local
|
| /external/tensorflow/tensorflow/python/autograph/lang/ |
| D | special_functions.py | 31 def _validate_list_constructor(elements, element_dtype, element_shape): argument
59 element_shape=None, argument
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | data_structures.py | 57 def tf_tensor_array_new(elements, element_dtype=None, element_shape=None): argument
107 def tf_tensor_list_new(elements, element_dtype=None, element_shape=None): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 79 TensorShape element_shape; in CreateZerosList() local
273 PartialTensorShape element_shape; in Compile() local
315 const TensorShape& element_shape, DataType dtype, in GetInitializedList()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape.cc | 50 for (const ShapeProto& element_shape : shape_proto.tuple_shapes()) { in Shape() local
|
| D | shape_util.cc | 395 for (const Shape& element_shape : shape.tuple_shapes()) { in ElementsInRecursive() local
406 for (const Shape& element_shape : shape.tuple_shapes()) { in HasPrimitiveType() local
632 for (auto& element_shape : shape.tuple_shapes()) { in ValidateShapeWithOptionalLayoutInternal() local
|
| D | layout_util.cc | 180 for (auto& element_shape : shape.tuple_shapes()) { in ValidateLayoutInShape() local
|
| D | xla.proto | 471 ShapeProto element_shape = 3; field
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 199 for (const Shape& element_shape : shape.tuple_shapes()) { in MakeFakeLiteralInternal() local
292 for (const Shape& element_shape : shape.tuple_shapes()) { in MakeFakeLiteralInternalWithBounds() local
|
| D | while_test.cc | 951 auto element_shape = ShapeUtil::MakeShape(F32, {2}); in XLA_TEST_F() local
983 auto element_shape = ShapeUtil::MakeShape(F32, {2}); in XLA_TEST_F() local
1008 auto element_shape = ShapeUtil::MakeShape(F32, {}); in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | padded_batch_dataset_op.cc | 287 const TensorShape& element_shape = in GetNextInternal() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion_test.cc | 1699 const Shape element_shape = ShapeUtil::MakeShape(F32, {42}); in TEST_F() local
1878 const Shape element_shape = ShapeUtil::MakeShape(F32, {42}); in MakeBenchmarkWhileBody() local
1995 const Shape element_shape = ShapeUtil::MakeShape(F32, {}); in MakeBenchmarkWhileBody() local
2014 const Shape element_shape = ShapeUtil::MakeShape(F32, {}); in BM_ManyElementTuple() local
|