| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 79 def __init__(self, iterator_resource, initializer, output_types, argument
102 if (output_types is None or output_shapes is None
108 output_types, output_shapes, output_classes)
120 def from_structure(output_types, argument
189 output_types = nest.map_structure(dtypes.as_dtype, output_types)
192 lambda _: tensor_shape.TensorShape(None), output_types)
194 output_shapes = nest.map_structure_up_to(output_types,
198 output_classes = nest.map_structure(lambda _: ops.Tensor, output_types)
199 nest.assert_same_structure(output_types, output_shapes)
201 output_types, output_shapes, output_classes)
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | sql_dataset_test.py | 42 output_types=(dtypes.string, dtypes.string, dtypes.string),
59 output_types=(dtypes.string, dtypes.string, dtypes.string)))
74 output_types=(dtypes.string, dtypes.string, dtypes.string)))
91 output_types=(dtypes.string, dtypes.string, dtypes.string)))
100 output_types=(dtypes.string, dtypes.string, dtypes.string)))
116 output_types=(dtypes.string, dtypes.string, dtypes.string)))
128 output_types=(dtypes.string, dtypes.string, dtypes.string))
138 output_types=(dtypes.string, dtypes.string, dtypes.string)))
149 output_types=(dtypes.string, dtypes.string, dtypes.string)))
161 output_types=(dtypes.string, dtypes.string, dtypes.string)))
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | op_version_test.cc | 75 .output_types = std::vector<TensorType>{TensorType_INT8}, in SimpleOutputVersioningTest()
82 .output_types = std::vector<TensorType>{TensorType_UINT8}, in SimpleOutputVersioningTest()
236 .output_types = std::vector<TensorType>{data_type}, in SimpleMulVersioningTest()
273 .output_types = std::vector<TensorType>{TensorType_UINT8}, in TEST()
282 .output_types = std::vector<TensorType>{TensorType_INT8}, in TEST()
320 .output_types = std::vector<TensorType>{TensorType_UINT8}, in TEST()
327 .output_types = std::vector<TensorType>{TensorType_INT8}, in TEST()
335 .output_types = std::vector<TensorType>{TensorType_FLOAT32}, in TEST()
376 .output_types = std::vector<TensorType>{TensorType_FLOAT32}, in TEST()
386 .output_types = std::vector<TensorType>{TensorType_FLOAT32}, in TEST()
[all …]
|
| D | op_version.cc | 36 op_sig.output_types.at(0) == TensorType_INT8) { in GetBuiltinOperatorVersion()
43 op_sig.output_types.at(0) == TensorType_FLOAT32) { in GetBuiltinOperatorVersion()
53 op_sig.output_types.at(0) == TensorType_FLOAT32) { in GetBuiltinOperatorVersion()
60 op_sig.output_types.at(0) == TensorType_INT8) { in GetBuiltinOperatorVersion()
96 op_sig.output_types.at(0) == TensorType_INT8) { in GetBuiltinOperatorVersion()
103 op_sig.output_types.at(0) == TensorType_FLOAT32) { in GetBuiltinOperatorVersion()
134 op_sig.output_types.at(0) == TensorType_FLOAT32) { in GetBuiltinOperatorVersion()
176 op_sig.output_types.at(0) == TensorType_FLOAT32) { in GetBuiltinOperatorVersion()
190 op_sig.output_types.at(0) == TensorType_FLOAT32) { in GetBuiltinOperatorVersion()
257 if (op_sig.output_types.at(0) == TensorType_INT8) { in GetBuiltinOperatorVersion()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MapDefun.pbtxt | 9 type_list_attr: "output_types"
18 name: "output_types"
46 type_list_attr: "output_types"
64 name: "output_types"
92 type_list_attr: "output_types"
110 name: "output_types"
|
| D | DatasetToSingleElement.pbtxt | 9 type_list_attr: "output_types"
12 name: "output_types"
32 type_list_attr: "output_types"
35 name: "output_types"
|
| D | ReduceDataset.pbtxt | 17 type_list_attr: "output_types"
35 name: "output_types"
70 type_list_attr: "output_types"
88 name: "output_types"
|
| D | PrefetchDataset.pbtxt | 16 name: "output_types"
44 name: "output_types"
71 name: "output_types"
105 name: "output_types"
|
| D | ParallelMapDataset.pbtxt | 29 name: "output_types"
70 name: "output_types"
110 name: "output_types"
157 name: "output_types"
211 name: "output_types"
|
| D | MapDataset.pbtxt | 25 name: "output_types"
62 name: "output_types"
98 name: "output_types"
141 name: "output_types"
|
| D | ModelDataset.pbtxt | 12 name: "output_types"
42 name: "output_types"
79 name: "output_types"
|
| D | RebatchDataset.pbtxt | 16 name: "output_types"
43 name: "output_types"
77 name: "output_types"
|
| D | ExperimentalRebatchDataset.pbtxt | 16 name: "output_types"
43 name: "output_types"
77 name: "output_types"
|
| D | ShuffleDataset.pbtxt | 24 name: "output_types"
60 name: "output_types"
102 name: "output_types"
|
| D | SnapshotDataset.pbtxt | 16 name: "output_types"
64 name: "output_types"
126 name: "output_types"
202 name: "output_types"
292 name: "output_types"
403 name: "output_types"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | decode_proto_ops.cc | 37 std::vector<tensorflow::DataType> output_types; in __anonc52216550102() local
38 TF_RETURN_IF_ERROR(c->GetAttr("output_types", &output_types)); in __anonc52216550102()
42 c->Concatenate(input, c->Vector(output_types.size()), &sizes)); in __anonc52216550102()
51 for (int i = 0; i < output_types.size(); ++i) { in __anonc52216550102()
|
| D | remote_fused_graph_ops_test.cc | 40 DataTypeVector output_types; in TEST() local
42 output_types.emplace_back(DT_FLOAT); in TEST()
47 .Attr("Toutputs", output_types); in TEST()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | from_generator_test.py | 40 generator, output_types=dtypes.int64).repeat(num_repeats).prefetch(5)
132 generator, output_types=(dtypes.int64, dtypes.int64),
186 generator, output_types=dtypes.int64, output_shapes=[]).prefetch(2)
207 generator, output_types=dtype, output_shapes=[1])
225 generator, output_types=dtypes.string, output_shapes=[])
238 generator, output_types=dtypes.int64, output_shapes=[3])
259 generator, output_types=dtypes.int64, output_shapes=[3])
280 generator, output_types=(dtypes.int64, dtypes.int64))
302 generator, output_types=dtypes.int64)
314 generator, output_types=dtypes.int64)
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | post_quantize.cc | 102 llvm::SmallVector<Type, 4> output_types; in RemoveQuantizationAdaptorOps() local
103 output_types.reserve(num_return_operands); in RemoveQuantizationAdaptorOps()
110 output_types.push_back(dequantized_result.getType()); in RemoveQuantizationAdaptorOps()
114 output_types.push_back(returned_value.getType()); in RemoveQuantizationAdaptorOps()
117 auto new_func_type = builder.getFunctionType(input_types, output_types); in RemoveQuantizationAdaptorOps()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.-iterator.pbtxt | 23 name: "output_types"
28 …argspec: "args=[\'self\', \'iterator_resource\', \'initializer\', \'output_types\', \'output_shape…
32 …argspec: "args=[\'string_handle\', \'output_types\', \'output_shapes\', \'output_classes\'], varar…
36 …argspec: "args=[\'output_types\', \'output_shapes\', \'shared_name\', \'output_classes\'], varargs…
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | tfprof_options.cc | 44 std::set<string> output_types(kOutput, in ParseOutput() local
49 if (output_types.find(output_opt) == output_types.end()) { in ParseOutput()
53 output_opt, absl::StrJoin(output_types, ","))); in ParseOutput()
58 if (output_types.find(*output_type) == output_types.end()) { in ParseOutput()
62 *output_type, absl::StrJoin(output_types, ","))); in ParseOutput()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | scan_ops.py | 57 if not (isinstance(wrapped_func.output_types, collections_abc.Sequence)
58 and len(wrapped_func.output_types) == 2):
79 new_state_types, output_types = wrapped_func.output_types
97 output_types, output_shapes, output_classes)
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | window_dataset.cc | 31 DataTypeVector output_types, in WindowDataset() argument
35 output_types_(std::move(output_types)),
127 DataTypeVector output_types, in NewWindowDataset() argument
132 *out_dataset = new WindowDataset(std::move(elements), std::move(output_types), in NewWindowDataset()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | insert_logging.cc | 113 DataTypeVector output_types; in InsertLogging() local
114 TF_RETURN_IF_ERROR(GetInOutTypes(node, &input_types, &output_types)); in InsertLogging()
130 SetNodeAttr("T", output_types[0], print_node); in InsertLogging()
134 SetNodeAttr("U", output_types, print_node); in InsertLogging()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils_test.cc | 178 auto output_types = fused_lstm_func_.getType().getResults(); in TEST_F() local
180 EXPECT_EQ(output_types[0].cast<RankedTensorType>().getShape().size(), in TEST_F()
183 EXPECT_EQ(output_types[0].cast<RankedTensorType>().getDimSize(i), in TEST_F()
247 auto output_types = fused_ln_lstm_func_.getType().getResults(); in TEST_F() local
249 EXPECT_EQ(output_types[0].cast<RankedTensorType>().getShape().size(), in TEST_F()
252 EXPECT_EQ(output_types[0].cast<RankedTensorType>().getDimSize(i), in TEST_F()
|