Home
last modified time | relevance | path

Searched refs:element_dtype (Results 1 – 25 of 79) sorted by relevance

1234

/external/tensorflow/tensorflow/core/ops/
Dlist_ops.cc29 DataType element_dtype) { in VerifyHandleData() argument
37 if (list_shape_type.dtype != element_dtype) { in VerifyHandleData()
39 DataTypeString(element_dtype), in VerifyHandleData()
65 DataType element_dtype; in __anonabedecb60202() local
66 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonabedecb60202()
72 {element_shape, element_dtype, ST_TENSOR_LIST}}); in __anonabedecb60202()
83 DataType element_dtype; in __anonabedecb60302() local
84 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonabedecb60302()
95 if (list_shape_type.dtype != element_dtype) { in __anonabedecb60302()
100 DataTypeString(element_dtype)); in __anonabedecb60302()
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dlist_ops_test.py56 element_dtype=dtypes.float32,
60 l, e = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
61 l = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
81 element_dtype=dtypes.float32, element_shape=[], max_num_elements=1)
93 element_dtype=dtypes.float32,
98 l = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
103 element_dtype=dtypes.float32, element_shape=[2, 3], num_elements=3)
104 _, e = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
105 l = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
112 element_dtype=dtypes.float32, element_shape=[None, 3], num_elements=3)
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dtensor_list_ops_test.py41 element_dtype=dtypes.float32,
52 element_dtype=dtypes.float32,
58 l, e2 = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
59 _, e1 = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
68 element_dtype=dtypes.float32,
77 l, e2 = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
78 _, e1 = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
86 element_dtype=dtypes.float32,
91 _, e11 = list_ops.tensor_list_pop_back(l, element_dtype=dtypes.float32)
92 l2, e21 = list_ops.tensor_list_pop_back(l2, element_dtype=dtypes.float32)
[all …]
Dadd_n_test.py35 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/
Ddata_structures.py57 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 …]
Ddata_structures_test.py45 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)
130 element_dtype=initial_list.dtype,
140 t = list_ops.tensor_list_stack(l, element_dtype=initial_list.dtype)
[all …]
Dslices_test.py36 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))
/external/tensorflow/tensorflow/python/ops/
Dlist_ops.py52 element_dtype, argument
60 element_dtype=element_dtype,
65 def _set_handle_data(list_handle, element_shape, element_dtype): argument
79 dtype=element_dtype.as_datatype_enum,
84 def tensor_list_reserve(element_shape, num_elements, element_dtype, name=None): argument
88 element_dtype=element_dtype,
92 _set_handle_data(result, element_shape, element_dtype)
106 def tensor_list_get_item(input_handle, index, element_dtype, element_shape=None, argument
112 element_dtype=element_dtype,
116 def tensor_list_pop_back(input_handle, element_dtype, name=None): argument
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dtensor_list.h72 element_dtype(other.element_dtype), in TensorList()
80 element_dtype(rhs.element_dtype), in TensorList()
89 element_dtype = rhs.element_dtype;
100 element_dtype = rhs.element_dtype;
119 DataType element_dtype; variable
133 out.element_dtype = element_dtype; in Copy()
Dlist_kernels.h72 c, element_dtype_ == tensor_list->element_dtype, in Compute()
75 " but list elements ", DataTypeString(tensor_list->element_dtype))); in Compute()
171 OP_REQUIRES(c, element_dtype_ == l->element_dtype, in Compute()
175 DataTypeString(l->element_dtype))); in Compute()
236 OP_REQUIRES(c, element_dtype_ == l->element_dtype, in Compute()
240 DataTypeString(l->element_dtype))); in Compute()
300 c, element_dtype_ == tensor_list->element_dtype, in Compute()
303 " but list elements ", DataTypeString(tensor_list->element_dtype))); in Compute()
490 output_list.element_dtype = input_tensor.dtype(); in Compute()
560 c, element_dtype_ == tensor_list->element_dtype, in Compute()
[all …]
Dlist_kernels.cc147 empty.element_dtype = element_dtype_; in Compute()
196 OP_REQUIRES(c, element_dtype_ == l->element_dtype, in Compute()
200 DataTypeString(l->element_dtype))); in Compute()
307 output.element_dtype = element_dtype_; in Compute()
365 output_list.element_dtype = input_list->element_dtype; in Compute()
403 OP_REQUIRES(c, element_dtype_ == l->element_dtype, in Compute()
407 DataTypeString(l->element_dtype))); in Compute()
511 OP_REQUIRES(c, l_a->element_dtype == element_dtype_, in Compute()
514 DataTypeString(l_a->element_dtype), " vs. ", in Compute()
516 OP_REQUIRES(c, l_b->element_dtype == element_dtype_, in Compute()
[all …]
Dtensor_list.cc46 core::PutVarint64(&metadata, static_cast<uint64>(element_dtype)); in Encode()
58 to->element_dtype = from.element_dtype; in TensorListDeviceCopy()
116 element_dtype = static_cast<DataType>(scratch); in Decode()
/external/tensorflow/tensorflow/python/autograph/lang/
Dspecial_functions.py31 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))
Dspecial_functions_test.py44 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/ops/compat/ops_history_v1/
DTensorListConcat.pbtxt9 type_attr: "element_dtype"
16 name: "element_dtype"
28 type_attr: "element_dtype"
35 name: "element_dtype"
DTensorListPushBack.pbtxt9 type_attr: "element_dtype"
16 name: "element_dtype"
DTensorListPushBackBatch.pbtxt9 type_attr: "element_dtype"
16 name: "element_dtype"
DTensorListGetItem.pbtxt17 type_attr: "element_dtype"
20 name: "element_dtype"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTensorListConcat.pbtxt9 type_attr: "element_dtype"
16 name: "element_dtype"
28 type_attr: "element_dtype"
35 name: "element_dtype"
DTensorListPushBackBatch.pbtxt9 type_attr: "element_dtype"
16 name: "element_dtype"
DTensorListPushBack.pbtxt9 type_attr: "element_dtype"
16 name: "element_dtype"
DTensorListScatterIntoExistingList.pbtxt9 type_attr: "element_dtype"
20 name: "element_dtype"
DTensorListPopBack.pbtxt17 type_attr: "element_dtype"
20 name: "element_dtype"
/external/llvm-project/mlir/test/mlir-tblgen/
Dop-derived-attribute.mlir5 // expected-remark @+2 {{element_dtype = f32}}
9 // expected-remark @+2 {{element_dtype = i79}}
13 // expected-remark @+2 {{element_dtype = complex<f32>}}
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dtensor-list.pbtxt136 key: "element_dtype"
156 key: "element_dtype"
177 key: "element_dtype"
191 key: "element_dtype"

1234