| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 30 DataType element_dtype) { in VerifyHandleData()
67 DataType element_dtype; in __anond381a6db0202() local
87 DataType element_dtype; in __anond381a6db0302() local
137 DataType element_dtype; in __anond381a6db0402() local
180 DataType element_dtype; in __anond381a6db0502() local
220 DataType element_dtype; in __anond381a6db0602() local
265 DataType element_dtype; in TensorListConcatShapeInference() local
338 DataType element_dtype; in __anond381a6db0902() local
374 DataType element_dtype; in __anond381a6db0a02() local
425 DataType element_dtype; in __anond381a6db0c02() local
[all …]
|
| /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
85 def tensor_list_reserve(element_shape, num_elements, element_dtype, name=None): argument
107 def tensor_list_get_item(input_handle, index, element_dtype, element_shape=None, argument
117 def tensor_list_pop_back(input_handle, element_dtype, name=None): argument
127 element_dtype, argument
162 element_dtype, argument
174 def tensor_list_concat(input_handle, element_dtype, element_shape=None, argument
|
| /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/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 | 513 Type element_dtype = op.element_dtype(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1074 if (auto element_dtype = node.attrs().Find("element_dtype")) { in InferOutputType() local
|