| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 67 shape_inference::ShapeHandle element_shape; in __anonabedecb60202() local
85 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anonabedecb60302() local
132 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anonabedecb60402() local
211 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anonabedecb60602() local
253 shape_inference::ShapeHandle element_shape) { in TensorListConcatShapeInference()
296 shape_inference::ShapeHandle element_shape; in __anonabedecb60702() local
311 shape_inference::ShapeHandle element_shape; in __anonabedecb60802() local
340 shape_inference::ShapeHandle element_shape; in __anonabedecb60902() local
366 shape_inference::ShapeHandle element_shape; in __anonabedecb60a02() local
406 shape_inference::ShapeHandle element_shape; in __anonabedecb60c02() local
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 51 def empty_tensor_list(element_shape, argument
65 def _set_handle_data(list_handle, element_shape, element_dtype): argument
84 def tensor_list_reserve(element_shape, num_elements, element_dtype, name=None): argument
96 def tensor_list_from_tensor(tensor, element_shape, name=None): argument
106 def tensor_list_get_item(input_handle, index, element_dtype, element_shape=None, argument
127 element_shape=None, argument
139 element_shape=None, argument
163 element_shape=None, argument
173 def tensor_list_concat(input_handle, element_dtype, element_shape=None, argument
185 def tensor_list_split(tensor, element_shape, lengths, name=None): argument
|
| D | tensor_array_ops.py | 65 element_shape=None, argument
183 def element_shape(self): member in _GraphTensorArray
406 element_shape=None, argument
485 def element_shape(self): member in _GraphTensorArrayV2
663 element_shape=None, argument
735 def element_shape(self): member in _EagerTensorArray
1011 element_shape=None, argument
1099 def element_shape(self): member in TensorArray
1324 def __init__(self, element_shape=None, dtype=dtypes.float32, argument
|
| /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/core/kernels/ |
| D | list_kernels.h | 108 TensorShape element_shape; in Compute() local
187 TensorShape element_shape; in Compute() local
252 TensorShape element_shape; in Compute() local
310 PartialTensorShape element_shape; in Compute() local
443 TensorShape element_shape = output_shape; in Compute() local
482 PartialTensorShape element_shape; in Compute() local
589 TensorShape element_shape; in Compute() local
657 PartialTensorShape element_shape; in Compute() local
778 PartialTensorShape element_shape; in Compute() local
|
| D | tensor_list.h | 117 PartialTensorShape element_shape; variable
|
| D | list_kernels.cc | 67 PartialTensorShape* element_shape) { in GetElementShapeFromInput()
149 PartialTensorShape element_shape; in Compute() local
302 PartialTensorShape element_shape; in Compute() local
|
| D | tensor_array.h | 139 int32 N, const PartialTensorShape& element_shape, in TensorArray()
561 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 | 169 TensorShape element_shape(tuple[component].shape()); in GetElementComponentFromBatch() local
|
| D | tensor_array_ops.cc | 332 auto element_shape = PartialTensorShape(); in CreateTensorArray() local
1039 TensorShape element_shape(tensor_value->shape()); in Compute() local
|
| /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 | 52 XlaOpKernelContext* ctx, const xla::Shape& element_shape, in GetTensorListDynamicDims()
166 xla::Shape element_shape; in Compile() local
238 xla::Shape element_shape; in Compile() local
486 xla::Shape element_shape = shape_or.ConsumeValueOrDie(); in Compile() local
532 xla::Shape element_shape = shape_or.ConsumeValueOrDie(); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_list_ops_test.py | 215 def testGather(self, input_list, element_shape, indices, output): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 144 const Shape& element_shape = return_shape.tuple_shapes(i); in CodegenNestedComputation() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape.cc | 50 for (const ShapeProto& element_shape : shape_proto.tuple_shapes()) { in Shape() local
|
| D | layout_util.cc | 182 for (auto& element_shape : shape.tuple_shapes()) { in ValidateLayoutInShape() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 97 const Shape& element_shape = tuple_shape.tuple_shapes(i); in EmitTupleAllocasAtFunctionEntry() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | collection_ops_util.cc | 135 LogicalResult CreateInitBufferValue(ArrayRef<int64_t> element_shape, in CreateInitBufferValue()
149 LogicalResult CreateInitBufferValue(ArrayRef<int64_t> element_shape, in CreateInitBufferValue()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 270 for (const Shape& element_shape : shape.tuple_shapes()) { in MakeFakeLiteralInternal() local
372 for (const Shape& element_shape : shape.tuple_shapes()) { in MakeFakeLiteralInternalWithBounds() local
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure.py | 56 def _TensorArrayStructure(dtype, element_shape, dynamic_size, infer_shape): argument
|