| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | list_ops.cc | 30 DataType element_dtype) { in VerifyHandleData()
68 DataType element_dtype; in __anon5e0c1ea50202() local
89 DataType element_dtype; in __anon5e0c1ea50302() local
140 DataType element_dtype; in __anon5e0c1ea50402() local
183 DataType element_dtype; in __anon5e0c1ea50502() local
223 DataType element_dtype; in __anon5e0c1ea50602() local
268 DataType element_dtype; in TensorListConcatShapeInference() local
342 DataType element_dtype; in __anon5e0c1ea50902() local
380 DataType element_dtype; in __anon5e0c1ea50a02() local
432 DataType element_dtype; in __anon5e0c1ea50c02() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | list_ops.py | 48 element_dtype, 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
103 def tensor_list_get_item(input_handle, index, element_dtype, element_shape=None, argument
113 def tensor_list_pop_back(input_handle, element_dtype, name=None): argument
123 element_dtype, argument
158 element_dtype, argument
170 def tensor_list_concat(input_handle, element_dtype, element_shape=None, argument
|
| /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
54 element_dtype=None, argument
88 def stack(list_or_tensor, element_dtype=None, strict=True): 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/mlir/tensorflow/transforms/ |
| H A D | collection_ops_util.cc | 137 Type element_dtype, OpBuilder builder, in CreateInitBufferValue()
151 Type element_dtype, OpBuilder builder, in CreateInitBufferValue()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | tensor_list.h | 119 DataType element_dtype; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/ |
| H A D | custom_call_encoding.cc | 875 DType element_dtype = ScalarDType(memref_ty.getElementType()); in EncodeMemRef() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | lower_static_tensor_list.cc | 583 Type element_dtype = op.element_dtype(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | import_model.cc | 1109 if (auto element_dtype = node.attrs().Find("element_dtype")) { in InferOutputType() local
|