Home
last modified time | relevance | path

Searched refs:dense_types (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/python/util/
Dexample_parser_configuration.py55 dense_types = parse_example_op.get_attr("Tdense")
63 if len(dense_types) != num_dense:
66 (len(dense_types), num_dense))
114 fixed_config.dtype = dense_types[i].as_datatype_enum
139 dense_types = parse_example_op.get_attr("Tdense")
146 num_dense = len(dense_types)
184 fixed_config.dtype = dense_types[i].as_datatype_enum
/external/tensorflow/tensorflow/python/ops/
Dparsing_config.py395 dense_types=None, argument
408 dense_types = [] if dense_types is None else dense_types
419 self.dense_types = [dtypes.as_dtype(t) for t in dense_types]
469 for k, s, t in zip(self.dense_keys, self.dense_shapes, self.dense_types)
579 self.dense_types.append(feature.dtype)
591 self.dense_types.append(feature.dtype)
628 if len(self.dense_types) != len(self.dense_keys):
631 (len(self.dense_types), len(self.dense_keys)))
Dparsing_ops.py645 feature_list_dense_types=feature_list.dense_types,
/external/tensorflow/tensorflow/core/ops/
Dragged_array_ops.cc67 std::vector<DataType> dense_types; in __anon9deb1dc20102() local
73 TF_RETURN_IF_ERROR(c->GetAttr("dense_types", &dense_types)); in __anon9deb1dc20102()
100 for (int i = 0; i < dense_types.size(); ++i) { in __anon9deb1dc20102()
Dops.pbtxt34761 type_list_attr: "dense_types"
34827 name: "dense_types"
48431 type_list_attr: "dense_types"
48475 name: "dense_types"
48524 type_list_attr: "dense_types"
48567 name: "dense_types"
48596 type_list_attr: "dense_types"
48631 name: "dense_types"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSparseCrossV2.pbtxt19 type_list_attr: "dense_types"
54 name: "dense_types"
DSparseCrossHashed.pbtxt19 type_list_attr: "dense_types"
62 name: "dense_types"
DSparseCross.pbtxt19 type_list_attr: "dense_types"
63 name: "dense_types"
DRaggedCross.pbtxt27 type_list_attr: "dense_types"
93 name: "dense_types"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DSparseCross.pbtxt19 type_list_attr: "dense_types"
63 name: "dense_types"
DRaggedCross.pbtxt27 type_list_attr: "dense_types"
93 name: "dense_types"
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.h167 TF_RETURN_IF_ERROR(ctx->GetAttr("Tdense", &dense_types));
193 std::vector<DataType> dense_types; member
213 TF_RETURN_IF_ERROR(ctx->GetAttr("Tdense", &dense_types)); in Init()
232 std::vector<DataType> dense_types; member
Dexample_proto_helper.cc414 num_dense = dense_types.size(); in FinishInit()
423 if (static_cast<size_t>(num_dense) != dense_types.size()) { in FinishInit()
440 for (const DataType& type : dense_types) { in FinishInit()
462 if (dense_keys.size() != dense_types.size()) { in FinishInit()
468 for (const DataType& type : dense_types) { in FinishInit()
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example_test.cc160 std::vector<TensorType> dense_types, in ParseExampleOpModel() argument
176 TensorData(dense_types[0], {dense_size}), dense_defaults); in ParseExampleOpModel()
190 dense_outputs_.push_back(AddOutput({dense_types[i], {dense_size}})); in ParseExampleOpModel()
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dparsing_ops.py65 self._dense_types = params.dense_types
76 params.dense_types,
/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops.cc196 if (def_value.dtype() != attrs_.dense_types[d]) { in CheckInputShapes()
201 "] == ", DataTypeString(attrs_.dense_types[d])); in CheckInputShapes()
216 config.dense.emplace_back(dense_keys_t[d], attrs_.dense_types[d], in MakeConfig()
343 OP_REQUIRES(ctx, def_value.dtype() == attrs_.dense_types[d], in Compute()
347 "] == ", DataTypeString(attrs_.dense_types[d]))); in Compute()
355 config.dense.push_back({attrs_.dense_keys[d], attrs_.dense_types[d], in Compute()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc218 const DataTypeVector& dense_types, in Dataset() argument
236 dense_types_(dense_types), in Dataset()
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt74804 type_list_attr: "dense_types"
74848 name: "dense_types"