| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 69 shape_inference::ShapeHandle element_shape; in __anond381a6db0202() local
89 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anond381a6db0302() local
139 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anond381a6db0402() local
222 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anond381a6db0602() local
264 shape_inference::ShapeHandle element_shape) { in TensorListConcatShapeInference()
307 shape_inference::ShapeHandle element_shape; in __anond381a6db0702() local
322 shape_inference::ShapeHandle element_shape; in __anond381a6db0802() local
352 shape_inference::ShapeHandle element_shape; in __anond381a6db0902() local
380 shape_inference::ShapeHandle element_shape; in __anond381a6db0a02() local
422 shape_inference::ShapeHandle element_shape; in __anond381a6db0c02() 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
85 def tensor_list_reserve(element_shape, num_elements, element_dtype, name=None): argument
97 def tensor_list_from_tensor(tensor, element_shape, name=None): argument
107 def tensor_list_get_item(input_handle, index, element_dtype, element_shape=None, argument
128 element_shape=None, argument
140 element_shape=None, argument
164 element_shape=None, argument
174 def tensor_list_concat(input_handle, element_dtype, element_shape=None, argument
186 def tensor_list_split(tensor, element_shape, lengths, name=None): argument
|
| D | tensor_array_ops.py | 64 element_shape=None, argument
190 def element_shape(self): member in _GraphTensorArray
413 element_shape=None, argument
496 def element_shape(self): member in _GraphTensorArrayV2
676 element_shape=None, argument
748 def element_shape(self): member in _EagerTensorArray
1025 element_shape=None, argument
1113 def element_shape(self): member in TensorArray
1352 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_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
1032 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 | 52 XlaOpKernelContext* ctx, const xla::Shape& element_shape, in GetTensorListDynamicDims()
168 xla::Shape element_shape; in Compile() local
240 xla::Shape element_shape; in Compile() local
488 xla::Shape element_shape = shape_or.ConsumeValueOrDie(); in Compile() local
534 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
|
| /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/llvm_ir/ |
| D | tuple_ops.cc | 97 const Shape& element_shape = tuple_shape.tuple_shapes(i); in EmitTupleAllocasAtFunctionEntry() 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
120 std::vector<int64> element_shape; in CreateSlice() local
135 std::vector<int64> element_shape; in CreateConcat() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 272 for (const Shape& element_shape : shape.tuple_shapes()) { in MakeFakeLiteralInternal() local
374 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
|