Home
last modified time | relevance | path

Searched refs:custom_options (Results 1 – 25 of 47) sorted by relevance

12

/external/libtextclassifier/native/utils/tflite/
Dstring_projection_base.cc53 const flexbuffers::Map& custom_options) in StringProjectionOpBase() argument
54 : hash_function_(custom_options["hash_function"].AsTypedVector()), in StringProjectionOpBase()
55 num_hash_(custom_options["num_hash"].AsInt32()), in StringProjectionOpBase()
56 num_bits_(custom_options["num_bits"].AsInt32()), in StringProjectionOpBase()
57 binary_projection_(custom_options["binary_projection"].AsBool()), in StringProjectionOpBase()
58 hash_method_(custom_options["hash_method"].ToString()), in StringProjectionOpBase()
59 axb_scale_(custom_options["axb_scale"].AsFloat()) { in StringProjectionOpBase()
61 .ngram_size = custom_options["ngram_size"].AsInt32(), in StringProjectionOpBase()
62 .max_skip_size = custom_options["max_skip_size"].AsInt32(), in StringProjectionOpBase()
63 .include_all_ngrams = custom_options["include_all_ngrams"].AsBool(), in StringProjectionOpBase()
[all …]
Dblacklist.cc35 explicit BlacklistOp(const flexbuffers::Map& custom_options) in BlacklistOp() argument
36 : BlacklistOpBase(custom_options), in BlacklistOp()
37 skipgram_finder_(custom_options["max_skip_size"].AsInt32()), in BlacklistOp()
39 auto blacklist = custom_options["blacklist"].AsTypedVector(); in BlacklistOp()
41 custom_options["blacklist_category"].AsTypedVector(); in BlacklistOp()
Dblacklist_base.h54 explicit BlacklistOpBase(const flexbuffers::Map& custom_options) in BlacklistOpBase() argument
55 : categories_(custom_options["categories"].AsInt32()), in BlacklistOpBase()
56 negative_categories_(custom_options["negative_categories"].AsInt32()) {} in BlacklistOpBase()
Dstring_projection.cc520 explicit StringProjectionOp(const flexbuffers::Map& custom_options) in StringProjectionOp() argument
521 : StringProjectionOpBase(custom_options), in StringProjectionOp()
523 custom_options["token_separators"].AsString().str(), in StringProjectionOp()
524 custom_options["normalize_repetition"].AsBool()), in StringProjectionOp()
526 if (custom_options["max_input_chars"].IsInt()) { in StringProjectionOp()
528 custom_options["max_input_chars"].AsInt32(); in StringProjectionOp()
Dstring_projection_base.h98 explicit StringProjectionOpBase(const flexbuffers::Map& custom_options);
/external/tensorflow/tensorflow/lite/toco/tflite/
Dcustom_operator.h58 const CustomOptions* custom_options) const override { in Deserialize() argument
60 if (custom_options) { in Deserialize()
62 flexbuffers::GetRoot(custom_options->data(), custom_options->size()) in Deserialize()
Dsimple_operator.h46 const CustomOptions* custom_options) const override { in Deserialize() argument
Dbuiltin_operator.h61 const CustomOptions* custom_options) const override { in Deserialize() argument
Dimport.cc127 input_op->custom_options()); in ImportOperators()
140 input_op->custom_options()); in ImportOperators()
Doperator.h89 const CustomOptions* custom_options) const = 0;
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_operator.cc240 const std::string& custom_code, const std::vector<uint8_t>& custom_options, in CustomOptionsToAttributes() argument
246 content.assign(reinterpret_cast<const char*>(custom_options.data()), in CustomOptionsToAttributes()
247 custom_options.size()); in CustomOptionsToAttributes()
249 {static_cast<int64_t>(custom_options.size())}, builder.getIntegerType(8)); in CustomOptionsToAttributes()
Dflatbuffer_operator.h69 const std::string &custom_code, const std::vector<uint8_t> &custom_options,
/external/tensorflow/tensorflow/lite/tools/
Dlist_flex_ops_test.cc74 const std::vector<uint8_t>& custom_options) { in FlexOpModel() argument
78 SetCustomOp(op_name, custom_options, Register_TEST); in FlexOpModel()
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib.cc118 flatbuffers::Offset<flatbuffers::Vector<uint8_t>> custom_options = 0; in ExportOperators() local
132 custom_writer->second(fbb, subgraph_, op_index, &custom_options, in ExportOperators()
136 custom_options = fbb->CreateVector( in ExportOperators()
151 custom_options, custom_options_format)); in ExportOperators()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dhashtable_resource.mlir25 // CHECK: custom_options:
Dtype_attr.mlir25 // CHECK: custom_options: [ 100, 116, 121, 112, 101, 0, 1, 7, 1, 1, 1, 2, 4, 2, 36, 1 ]
Dflex_exclusively.mlir32 // CHECK-NEXT: custom_options: [ 5, 65, 100, 100, 86, 50, 0, 22, 18, 5, 65, 100, 100, 86, 50…
/external/perfetto/protos/perfetto/metrics/chrome/
Dtest_chrome_metric.proto21 import "protos/perfetto/metrics/custom_options.proto";
Dframe_times.proto21 import "protos/perfetto/metrics/custom_options.proto";
Dreported_by_page.proto21 import "protos/perfetto/metrics/custom_options.proto";
Dscroll_jank.proto21 import "protos/perfetto/metrics/custom_options.proto";
Dmedia_metric.proto21 import "protos/perfetto/metrics/custom_options.proto";
/external/tensorflow/tensorflow/lite/
Dinterpreter_builder.cc319 if (op_type != BuiltinOperator_CUSTOM && op->custom_options()) { in ParseNodes()
326 if (op->custom_options()) { in ParseNodes()
331 reinterpret_cast<const char*>(op->custom_options()->data()), in ParseNodes()
332 op->custom_options()->size(), nullptr, registration); in ParseNodes()
/external/perfetto/protos/perfetto/metrics/
DBUILD.gn29 sources = [ "custom_options.proto" ]
/external/tensorflow/tensorflow/lite/kernels/
Dnumeric_verify_test.cc55 std::vector<uint8_t> custom_options(sizeof(float)); in NumericVerifyOpModel() local

12