Home
last modified time | relevance | path

Searched refs:sparse_types (Results 1 – 25 of 50) sorted by relevance

12

/external/tensorflow/tensorflow/python/util/
Dexample_parser_configuration.py54 sparse_types = parse_example_op.get_attr("sparse_types")
58 if len(sparse_types) != num_sparse:
61 (len(sparse_types), num_sparse))
124 var_len_feature.dtype = sparse_types[i].as_datatype_enum
141 sparse_types = parse_example_op.get_attr("sparse_types")
194 var_len_feature.dtype = sparse_types[i].as_datatype_enum
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.cc65 auto sparse_types = attr_map.at("sparse_types"); in ExtractExampleParserConfiguration() local
80 if (sparse_types.list().type_size() != num_sparse) { in ExtractExampleParserConfiguration()
82 sparse_types.list().type_size(), in ExtractExampleParserConfiguration()
92 for (int i = 0; i < sparse_types.list().type_size(); ++i) { in ExtractExampleParserConfiguration()
93 (*var_len_features)[i].dtype = sparse_types.list().type(i); in ExtractExampleParserConfiguration()
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example_test.cc64 key: "sparse_types"
92 key: "sparse_types"
120 key: "sparse_types"
148 key: "sparse_types"
161 std::vector<TensorType> sparse_types, in ParseExampleOpModel() argument
183 sparse_values_outputs_.push_back(AddOutput(sparse_types[i])); in ParseExampleOpModel()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.h166 TF_RETURN_IF_ERROR(ctx->GetAttr("sparse_types", &sparse_types));
192 std::vector<DataType> sparse_types; member
211 TF_RETURN_IF_ERROR(ctx->GetAttr("sparse_types", &sparse_types)); in Init()
218 if (num_sparse != sparse_keys.size() || num_sparse != sparse_types.size()) { in Init()
221 sparse_keys.size(), ") and sparse_types (", sparse_types.size(), ")"); in Init()
230 std::vector<DataType> sparse_types; member
/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops_test.cc163 std::vector<DataType> sparse_types; in ParseExample() local
184 sparse_types.push_back(opt.filler.dtype); in ParseExample()
196 .Attr("sparse_types", sparse_types) in ParseExample()
224 std::vector<DataType> sparse_types; in ParseExampleV2() local
245 sparse_types.push_back(opt.filler.dtype); in ParseExampleV2()
271 .Attr("sparse_types", sparse_types) in ParseExampleV2()
293 std::vector<DataType> sparse_types; in ParseSingleExample() local
312 sparse_keys.push_back(key), sparse_types.push_back(opt.filler.dtype); in ParseSingleExample()
323 .Attr("sparse_types", sparse_types) in ParseSingleExample()
/external/tensorflow/tensorflow/python/ops/
Dparsing_config.py393 sparse_types=None, argument
406 sparse_types = [] if sparse_types is None else sparse_types
417 self.sparse_types = [dtypes.as_dtype(t) for t in sparse_types]
534 original_dtype = self.sparse_types[self.sparse_keys.index(key)]
540 self.sparse_types.append(dtype)
632 if len(self.sparse_types) != len(self.sparse_keys):
635 (len(self.sparse_types), len(self.sparse_keys)))
Dparsing_ops.py358 sparse_types=params.sparse_types,
642 context_sparse_types=context.sparse_types,
646 feature_list_sparse_types=feature_list.sparse_types,
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSparseCrossV2.pbtxt10 type_list_attr: "sparse_types"
43 name: "sparse_types"
DSparseCrossHashed.pbtxt10 type_list_attr: "sparse_types"
51 name: "sparse_types"
DParseSingleExample.pbtxt18 type_list_attr: "sparse_types"
45 name: "sparse_types"
DParseExample.pbtxt32 type_list_attr: "sparse_types"
54 name: "sparse_types"
DSparseCross.pbtxt10 type_list_attr: "sparse_types"
52 name: "sparse_types"
DParseExampleV2.pbtxt34 type_list_attr: "sparse_types"
71 name: "sparse_types"
DExperimentalParseExampleDataset.pbtxt30 name: "sparse_types"
100 name: "sparse_types"
DParseExampleDatasetV2.pbtxt30 name: "sparse_types"
/external/tensorflow/tensorflow/core/kernels/fuzzing/
Dexample_proto_fast_parsing_fuzz.cc42 DataTypeSlice sparse_types = {}; in BuildGraph() local
47 sparse_keys, dense_keys, dense_defaults, sparse_types, in BuildGraph()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DParseSingleExample.pbtxt18 type_list_attr: "sparse_types"
45 name: "sparse_types"
DParseExample.pbtxt32 type_list_attr: "sparse_types"
54 name: "sparse_types"
DSparseCross.pbtxt10 type_list_attr: "sparse_types"
52 name: "sparse_types"
DParseExampleV2.pbtxt34 type_list_attr: "sparse_types"
71 name: "sparse_types"
DExperimentalParseExampleDataset.pbtxt30 name: "sparse_types"
100 name: "sparse_types"
DParseExampleDataset.pbtxt30 name: "sparse_types"
107 name: "sparse_types"
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dparsing_ops.py58 self._sparse_types = params.sparse_types
71 for (key, value_type) in zip(params.sparse_keys, params.sparse_types):
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ParseSingleExample.pbtxt28 must match the lengths of `sparse_keys` and `sparse_types`.
46 name: "sparse_types"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dparse_example_v2.mlir14 …[], result_segment_sizes = dense<[2, 2, 2, 2, 0, 0]> : vector<6xi32>, sparse_types = ["tfdtype$DT_…
65 // CHECK-NEXT: key: "sparse_types"

12