| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 29 DataType element_dtype) { in VerifyHandleData()
65 DataType element_dtype; in __anonabedecb60202() local
83 DataType element_dtype; in __anonabedecb60302() local
130 DataType element_dtype; in __anonabedecb60402() local
172 DataType element_dtype; in __anonabedecb60502() local
209 DataType element_dtype; in __anonabedecb60602() local
254 DataType element_dtype; in TensorListConcatShapeInference() local
326 DataType element_dtype; in __anonabedecb60902() local
360 DataType element_dtype; in __anonabedecb60a02() local
409 DataType element_dtype; in __anonabedecb60c02() local
[all …]
|
| /external/tensorflow/tensorflow/python/autograph/lang/ |
| D | special_functions.py | 31 def _validate_list_constructor(elements, element_dtype, element_shape): argument
58 element_dtype=None, argument
92 def stack(list_or_tensor, element_dtype=None, strict=True): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 52 element_dtype, argument
65 def _set_handle_data(list_handle, element_shape, element_dtype): argument
84 def tensor_list_reserve(element_shape, num_elements, element_dtype, name=None): argument
106 def tensor_list_get_item(input_handle, index, element_dtype, element_shape=None, argument
116 def tensor_list_pop_back(input_handle, element_dtype, name=None): argument
126 element_dtype, argument
161 element_dtype, argument
173 def tensor_list_concat(input_handle, element_dtype, 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/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/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 439 Type element_dtype = op.element_dtype(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 979 if (auto element_dtype = node.attrs().Find("element_dtype")) { in InferOutputType() local
|