| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 30 DataType element_dtype) { in VerifyHandleData()
68 DataType element_dtype; in __anon6c1e81940202() local
89 DataType element_dtype; in __anon6c1e81940302() local
140 DataType element_dtype; in __anon6c1e81940402() local
183 DataType element_dtype; in __anon6c1e81940502() local
223 DataType element_dtype; in __anon6c1e81940602() local
268 DataType element_dtype; in TensorListConcatShapeInference() local
342 DataType element_dtype; in __anon6c1e81940902() local
380 DataType element_dtype; in __anon6c1e81940a02() local
432 DataType element_dtype; in __anon6c1e81940c02() local
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| 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
|
| /external/tensorflow/tensorflow/python/autograph/lang/ |
| 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
|
| /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/core/kernels/ |
| D | tensor_list.h | 119 DataType element_dtype; variable
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | collection_ops_util.cc | 137 Type element_dtype, OpBuilder builder, in CreateInitBufferValue()
151 Type element_dtype, OpBuilder builder, in CreateInitBufferValue()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/ |
| D | custom_call_encoding.cc | 875 DType element_dtype = ScalarDType(memref_ty.getElementType()); in EncodeMemRef() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 583 Type element_dtype = op.element_dtype(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1109 if (auto element_dtype = node.attrs().Find("element_dtype")) { in InferOutputType() local
|