| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | list_ops.cc | 70 shape_inference::ShapeHandle element_shape; in __anon5e0c1ea50202() local
91 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anon5e0c1ea50302() local
142 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anon5e0c1ea50402() local
225 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anon5e0c1ea50602() local
267 shape_inference::ShapeHandle element_shape) { in TensorListConcatShapeInference()
310 shape_inference::ShapeHandle element_shape; in __anon5e0c1ea50702() local
325 shape_inference::ShapeHandle element_shape; in __anon5e0c1ea50802() local
356 shape_inference::ShapeHandle element_shape; in __anon5e0c1ea50902() local
386 shape_inference::ShapeHandle element_shape; in __anon5e0c1ea50a02() local
429 shape_inference::ShapeHandle element_shape; in __anon5e0c1ea50c02() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A 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
|
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A 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
|
| H A D | list_kernels.cc | 72 PartialTensorShape* element_shape) { in GetElementShapeFromInput()
154 PartialTensorShape element_shape; in Compute() local
325 PartialTensorShape element_shape; in Compute() local
|
| H A D | tensor_array.h | 139 int32_t N, const PartialTensorShape& element_shape, in TensorArray()
557 TensorShape element_shape; in LockedRead() local
|
| H A D | barrier_ops.cc | 93 TensorShape element_shape = values.shape(); in TryInsertMany() local
308 const Tensor& values, const TensorShape& element_shape, in InsertOneLocked()
|
| H A D | tensor_list.h | 117 PartialTensorShape element_shape; variable
|
| H A D | tensor_array_ops.cc | 332 auto element_shape = PartialTensorShape(); in CreateTensorArray() local
1033 TensorShape element_shape(tensor_value->shape()); in Compute() local
|
| H A D | deserialize_sparse_variant_op.cc | 102 const Tensor* element_shape; in Compute() local
|
| H A D | fifo_queue.cc | 93 TensorShape element_shape(tuple[component].shape()); in GetElementComponentFromBatch() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/lang/ |
| H A D | special_functions.py | 27 def _validate_list_constructor(elements, element_dtype, element_shape): argument
55 element_shape=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A 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
|
| H A D | copy_insertion_test.cc | 1748 const Shape element_shape = ShapeUtil::MakeShape(F32, {42}); in TEST_F() local
1927 const Shape element_shape = ShapeUtil::MakeShape(F32, {42}); in MakeBenchmarkWhileBody() local
2051 const Shape element_shape = ShapeUtil::MakeShape(F32, {}); in MakeBenchmarkWhileBody() local
2070 const Shape element_shape = ShapeUtil::MakeShape(F32, {}); in BM_ManyElementTuple() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A 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
|
| H A D | while_test.cc | 936 auto element_shape = ShapeUtil::MakeShape(F32, {2}); in XLA_TEST_F() local
968 auto element_shape = ShapeUtil::MakeShape(F32, {2}); in XLA_TEST_F() local
993 auto element_shape = ShapeUtil::MakeShape(F32, {}); in XLA_TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | lower_static_tensor_list.cc | 475 Value element_shape = adaptor.getOperands()[0]; in matchAndRewrite() local
917 Value element_shape = adaptor.getOperands()[1]; in matchAndRewrite() local
965 Value element_shape = adaptor.getOperands()[1]; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | tensor_list_ops_test.py | 212 def testGather(self, input_list, element_shape, indices, output): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | tuple_ops.cc | 75 const Shape& element_shape = tuple_shape.tuple_shapes(i); in EmitTupleAllocasAtFunctionEntry() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | shape.cc | 50 for (const ShapeProto& element_shape : shape_proto.tuple_shapes()) { in Shape() local
|
| H A D | shape_util.cc | 614 for (const Shape& element_shape : shape.tuple_shapes()) { in ElementsInRecursive() local
625 for (const Shape& element_shape : shape.tuple_shapes()) { in HasPrimitiveType() local
869 for (auto& element_shape : shape.tuple_shapes()) { in ValidateShapeWithOptionalLayoutInternal() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_nested.cc | 155 const Shape& element_shape = return_shape.tuple_shapes(i); in CodegenNestedComputation() local
|