| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 28 DataType element_dtype) { in VerifyHandleData() argument
36 if (list_shape_type.dtype != element_dtype) { in VerifyHandleData()
38 DataTypeString(element_dtype), in VerifyHandleData()
59 DataType element_dtype; in __anon9187032f0202() local
60 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anon9187032f0202()
66 {element_shape, element_dtype}}); in __anon9187032f0202()
77 DataType element_dtype; in __anon9187032f0302() local
78 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anon9187032f0302()
89 if (list_shape_type.dtype != element_dtype) { in __anon9187032f0302()
94 DataTypeString(element_dtype)); in __anon9187032f0302()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | list_ops_test.py | 50 element_dtype=dtypes.float32,
54 l, e = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
73 element_dtype=dtypes.float32, element_shape=[], max_num_elements=1)
85 element_dtype=dtypes.float32,
90 l = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
95 element_dtype=dtypes.float32, element_shape=[2, 3], num_elements=3)
96 _, e = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
101 element_dtype=dtypes.float32, element_shape=[None, 3], num_elements=3)
103 l, element_dtype=dtypes.float32, element_shape=[4, 3])
108 element_dtype=dtypes.float32, element_shape=None, num_elements=3)
[all …]
|
| D | while_v2_test.py | 197 element_dtype=x.dtype, element_shape=x.shape)
223 stack = list_ops.tensor_list_stack(outputs[1], element_dtype=x.dtype)
257 element_dtype=dtypes.float32, element_shape=ScalarShape())
280 element_dtype=dtypes.float32, element_shape=ScalarShape())
356 element_dtype=dtypes.float32)
369 element_dtype=dtypes.float32)
|
| D | resource_variable_ops_test.py | 973 element_dtype=dtypes.float32, element_shape=[])
979 list_ops.tensor_list_get_item(v[0], 0, element_dtype=dtypes.float32),
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_list_ops_test.py | 39 element_dtype=dtypes.float32,
50 element_dtype=dtypes.float32,
56 l, e2 = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
57 _, e1 = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
66 element_dtype=dtypes.float32,
75 l, e2 = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
76 _, e1 = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
84 element_dtype=dtypes.float32,
89 _, e11 = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
90 l2, e21 = list_ops.tensor_list_pop_back(l2, element_dtype=dtypes.float32)
[all …]
|
| D | add_n_test.py | 35 element_shape=[], element_dtype=dtypes.float32, num_elements=3)
37 element_shape=[], element_dtype=dtypes.float32, num_elements=3)
43 list_ops.tensor_list_stack(l, element_dtype=dtypes.float32),
49 element_shape=[], element_dtype=dtypes.float32, num_elements=2)
51 element_shape=[], element_dtype=dtypes.float32, num_elements=3)
56 list_ops.tensor_list_stack(l, element_dtype=dtypes.float32).eval()
66 element_dtype=dtypes.float32,
70 element_dtype=dtypes.float32,
77 list_ops.tensor_list_stack(l, element_dtype=dtypes.float32), {
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | data_structures.py | 57 def tf_tensor_array_new(elements, element_dtype=None, element_shape=None): argument
64 if element_dtype is not None and element_dtype != inferred_dtype:
67 element_dtype, elements, inferred_dtype))
73 if element_dtype is None:
91 if element_dtype is None:
92 element_dtype = inferred_dtype
97 dtype=element_dtype,
107 def tf_tensor_list_new(elements, element_dtype=None, element_shape=None): argument
122 if element_dtype is not None and element_dtype != inferred_dtype:
125 element_dtype, elements, inferred_dtype))
[all …]
|
| D | data_structures_test.py | 45 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
51 element_dtype=dtypes.int32,
53 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
59 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
69 data_structures.tf_tensor_list_new([3, 4], element_dtype=dtypes.float32)
88 data_structures.tf_tensor_array_new([3, 4], element_dtype=dtypes.float32)
97 data_structures.tf_tensor_array_new([], element_dtype=dtypes.float32),
105 t = list_ops.tensor_list_stack(l, element_dtype=x.dtype)
129 element_dtype=initial_list.dtype,
139 t = list_ops.tensor_list_stack(l, element_dtype=initial_list.dtype)
[all …]
|
| D | slices_test.py | 36 t = list_ops.tensor_list_stack(l, element_dtype=initial_list.dtype)
44 l, 1, slices.GetItemOpts(element_dtype=initial_list.dtype))
52 slices.GetItemOpts(element_dtype=initial_str.dtype))
59 slices.GetItemOpts(element_dtype=initial_str.dtype))
|
| D | slices.py | 76 if opts.element_dtype is None:
79 x = list_ops.tensor_list_get_item(target, i, element_dtype=opts.element_dtype)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 46 element_dtype, argument
54 element_dtype=element_dtype,
59 def tensor_list_reserve(element_shape, num_elements, element_dtype, name=None): argument
63 element_dtype=element_dtype,
74 def tensor_list_get_item(input_handle, index, element_dtype, element_shape=None, argument
80 element_dtype=element_dtype,
84 def tensor_list_pop_back(input_handle, element_dtype, name=None): argument
88 element_dtype=element_dtype,
94 element_dtype, argument
101 element_dtype=element_dtype,
[all …]
|
| D | tensor_array_ops.py | 491 element_dtype=dtype,
553 element_dtype=self._dtype,
584 element_dtype=self._dtype,
599 element_dtype=self._dtype,
615 element_dtype=self._dtype,
|
| D | while_v2.py | 767 element_dtype=tensor.dtype, element_shape=tensor.shape,
792 captured_accumulator, element_dtype=tensor.dtype)
|
| D | gradients_test.py | 1126 element_dtype=dtypes.float32,
1132 element_dtype=dtypes.float32,
|
| /external/tensorflow/tensorflow/python/autograph/lang/ |
| D | special_functions.py | 31 def _validate_list_constructor(elements, element_dtype, element_shape): argument
33 if element_dtype is not None and element_shape is not None:
58 element_dtype=None, argument
83 _validate_list_constructor(elements, element_dtype, element_shape)
85 return data_structures.tf_tensor_array_new(elements, element_dtype,
88 return data_structures.tf_tensor_list_new(elements, element_dtype,
92 def stack(list_or_tensor, element_dtype=None, strict=True): argument
122 element_dtype=element_dtype, original_call=original_call))
|
| D | special_functions_test.py | 44 element_dtype=dtypes.int32,
46 sl = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
51 element_dtype=dtypes.int32,
53 sl = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
60 sl = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
77 sl = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
108 special_functions.stack(l, element_dtype=dtypes.float32)))
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.cc | 45 element_dtype(other.element_dtype), in TensorList()
66 core::PutVarint64(&metadata, static_cast<uint64>(element_dtype)); in Encode()
78 to->element_dtype = from.element_dtype; in TensorListDeviceCopy()
136 element_dtype = static_cast<DataType>(scratch); in Decode()
237 empty.element_dtype = element_dtype_; in Compute()
288 OP_REQUIRES(c, element_dtype_ == l->element_dtype, in Compute()
292 DataTypeString(l->element_dtype))); in Compute()
390 output.element_dtype = element_dtype_; in Compute()
440 output_list.element_dtype = input_list->element_dtype; in Compute()
479 OP_REQUIRES(c, element_dtype_ == l->element_dtype, in Compute()
[all …]
|
| D | list_kernels.h | 61 DataType element_dtype; member
94 c, element_dtype_ == tensor_list->element_dtype, in Compute()
97 " but list elements ", DataTypeString(tensor_list->element_dtype))); in Compute()
192 OP_REQUIRES(c, element_dtype_ == l->element_dtype, in Compute()
196 DataTypeString(l->element_dtype))); in Compute()
257 OP_REQUIRES(c, element_dtype_ == l->element_dtype, in Compute()
261 DataTypeString(l->element_dtype))); in Compute()
324 c, element_dtype_ == tensor_list->element_dtype, in Compute()
327 " but list elements ", DataTypeString(tensor_list->element_dtype))); in Compute()
513 output_list.element_dtype = input_tensor.dtype(); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EmptyTensorList.pbtxt | 5 All list elements must be tensors of dtype element_dtype and shape compatible
9 element_dtype: the type of elements in the list.
|
| D | api_def_TensorListReserve.pbtxt | 8 element_dtype: the desired type of elements in the list.
|
| D | api_def_TensorListPopBack.pbtxt | 9 element_dtype: the type of elements in the list
|
| D | api_def_TensorListPushBack.pbtxt | 8 element_dtype: the type of elements in the list.
|
| /external/tensorflow/tensorflow/python/autograph/utils/ |
| D | tensors_test.py | 36 element_shape=constant_op.constant([1]), element_dtype=dtypes.int32)
|
| D | tensor_list_test.py | 45 s = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
|
| /external/tensorflow/tensorflow/contrib/autograph/examples/notebooks/ |
| D | algorithms.ipynb | 1383 " element_dtype=None))\n",
1385 " element_dtype=None))\n",
1391 " (ni), opts=ag__.GetItemOpts(element_dtype=None)),\n",
1392 " (nj), opts=ag__.GetItemOpts(element_dtype=None))\n",
1403 " ag__.GetItemOpts(element_dtype=None))))))), if_true,\n",
1414 " =ag__.GetItemOpts(element_dtype=None)), (j), opts=\n",
1415 " ag__.GetItemOpts(element_dtype=None))\n",
1440 " opts=ag__.GetItemOpts(element_dtype=None)))), extra_test_1,\n",
1447 " element_dtype=tf.int32, original_call=ag.stack))\n",
1471 " element_dtype=tf.int32, original_call=ag.stack))\n",
|