| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 70 shape_inference::ShapeHandle element_shape; in __anon6c1e81940202() local
91 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anon6c1e81940302() local
142 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anon6c1e81940402() local
225 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anon6c1e81940602() local
267 shape_inference::ShapeHandle element_shape) { in TensorListConcatShapeInference()
310 shape_inference::ShapeHandle element_shape; in __anon6c1e81940702() local
325 shape_inference::ShapeHandle element_shape; in __anon6c1e81940802() local
356 shape_inference::ShapeHandle element_shape; in __anon6c1e81940902() local
386 shape_inference::ShapeHandle element_shape; in __anon6c1e81940a02() local
429 shape_inference::ShapeHandle element_shape; in __anon6c1e81940c02() local
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 47 def empty_tensor_list(element_shape, argument
61 def _set_handle_data(list_handle, element_shape, element_dtype): argument
81 def tensor_list_reserve(element_shape, num_elements, element_dtype, name=None): argument
93 def tensor_list_from_tensor(tensor, element_shape, name=None): argument
103 def tensor_list_get_item(input_handle, index, element_dtype, element_shape=None, argument
124 element_shape=None, argument
136 element_shape=None, argument
160 element_shape=None, argument
170 def tensor_list_concat(input_handle, element_dtype, element_shape=None, argument
182 def tensor_list_split(tensor, element_shape, lengths, name=None): argument
|
| D | tensor_array_ops.py | 60 element_shape=None, argument
186 def element_shape(self): member in _GraphTensorArray
409 element_shape=None, argument
492 def element_shape(self): member in _GraphTensorArrayV2
672 element_shape=None, argument
744 def element_shape(self): member in _EagerTensorArray
1028 element_shape=None, argument
1116 def element_shape(self): member in TensorArray
1355 element_shape=None, argument
|
| /external/tensorflow/tensorflow/python/autograph/lang/ |
| D | special_functions.py | 27 def _validate_list_constructor(elements, element_dtype, element_shape): argument
55 element_shape=None, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 213 TensorShape element_shape; in Compute() local
295 TensorShape element_shape; in Compute() local
359 TensorShape element_shape; in Compute() local
416 PartialTensorShape element_shape; in Compute() local
548 TensorShape element_shape = output_shape; in Compute() local
590 PartialTensorShape element_shape; in Compute() local
701 TensorShape element_shape; in Compute() local
772 PartialTensorShape element_shape; in Compute() local
903 PartialTensorShape element_shape; in Compute() local
|
| D | tensor_list.h | 117 PartialTensorShape element_shape; variable
|
| D | list_kernels.cc | 72 PartialTensorShape* element_shape) { in GetElementShapeFromInput()
154 PartialTensorShape element_shape; in Compute() local
325 PartialTensorShape element_shape; in Compute() local
|
| D | tensor_array.h | 139 int32_t N, const PartialTensorShape& element_shape, in TensorArray()
557 TensorShape element_shape; in LockedRead() local
|
| D | deserialize_sparse_variant_op.cc | 102 const Tensor* element_shape; in Compute() local
|
| D | barrier_ops.cc | 93 TensorShape element_shape = values.shape(); in TryInsertMany() local
308 const Tensor& values, const TensorShape& element_shape, in InsertOneLocked()
|
| D | fifo_queue.cc | 93 TensorShape element_shape(tuple[component].shape()); in GetElementComponentFromBatch() local
|
| D | priority_queue.cc | 121 TensorShape element_shape(tuple[component].shape()); in GetElementComponentFromBatch() local
|
| D | padding_fifo_queue.cc | 62 TensorShape element_shape(tuple[component].shape()); in GetElementComponent() local
|
| D | random_shuffle_queue_op.cc | 172 TensorShape element_shape(tuple[component].shape()); in GetElementComponentFromBatch() local
|
| D | tensor_array_ops.cc | 332 auto element_shape = PartialTensorShape(); in CreateTensorArray() local
1033 TensorShape element_shape(tensor_value->shape()); in Compute() local
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | data_structures.py | 53 def tf_tensor_array_new(elements, element_dtype=None, element_shape=None): argument
103 def tf_tensor_list_new(elements, element_dtype=None, element_shape=None): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 53 XlaOpKernelContext* ctx, const xla::Shape& element_shape, in GetTensorListDynamicDims()
170 xla::Shape element_shape; in Compile() local
244 xla::Shape element_shape; in Compile() local
494 xla::Shape element_shape = std::move(shape_or).value(); in Compile() local
540 xla::Shape element_shape = std::move(shape_or).value(); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_list_ops_test.py | 212 def testGather(self, input_list, element_shape, indices, output): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 75 const Shape& element_shape = tuple_shape.tuple_shapes(i); in EmitTupleAllocasAtFunctionEntry() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape.cc | 50 for (const ShapeProto& element_shape : shape_proto.tuple_shapes()) { in Shape() 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/service/gpu/ |
| D | ir_emitter_nested.cc | 155 const Shape& element_shape = return_shape.tuple_shapes(i); in CodegenNestedComputation() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_concat_code_motion.cc | 83 const Shape& element_shape = elements.back()->shape(); in GetConcatShape() local
121 std::vector<int64_t> element_shape; in CreateSlice() local
136 std::vector<int64_t> element_shape; in CreateConcat() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 274 for (const Shape& element_shape : shape_tuple_shapes) { in MakeFakeLiteralInternal() local
378 for (const Shape& element_shape : shape_tuple_shapes) { in MakeFakeLiteralInternalWithBounds() local
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure.py | 53 def _TensorArrayStructure(dtype, element_shape, dynamic_size, infer_shape): argument
|