Home
last modified time | relevance | path

Searched refs:weights_format (Results 1 – 25 of 46) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dexternal_constant.mlir7 …1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
12 …%[[CONST]]) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"}
Dimporter_test_min_max.cc.mlir7 …1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
12 …%[[CONST]]) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"}
Doptional.mlir9 …1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dfully_connected_v2.mlir49 // CHECK-NEXT: weights_format: SHUFFLED4x16INT8
76 …1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "SHUFFLED4x1…
Doptional.mlir5 …rg1, %0) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
Dsignature_def.mlir113 …0, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
114 …1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
Dfully_connected.mlir76 …1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dshuffle_fc_weights.cc36 if (fc_op->weights_format != FullyConnectedWeightsFormat::kDefault) { in Run()
140 fc_op->weights_format = FullyConnectedWeightsFormat::kShuffled4x16Int8; in Run()
Densure_uint8_weights_safe_for_fast_int8_kernels.cc126 CHECK(fc_op.weights_format == FullyConnectedWeightsFormat::kDefault) in Run()
Didentify_lstm.cc247 if (static_cast<FullyConnectedOperator*>(fully_connected)->weights_format != in Run()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/
Dunroll_batch_matmul.pbtxt97 …VAL_5]]) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
98 …VAL_5]]) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.h40 FullyConnectedOptionsWeightsFormat weights_format; member
Dop_version.cc169 if (op_sig.options.fully_connected.weights_format == in GetBuiltinOperatorVersion()
714 op_sig.options.fully_connected.weights_format = in GetOpSignature()
715 fully_connected_option->weights_format(); in GetOpSignature()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dop_version_test.cc91 fc->weights_format = FullyConnectedWeightsFormat::kShuffled4x16Int8; in TEST()
136 fc->weights_format = FullyConnectedWeightsFormat::kShuffled4x16Int8; in TEST()
Doperator.cc465 *builder, activation_function, GetWeightFormat(op.weights_format)); in WriteOptions()
472 switch (options.weights_format()) { in ReadOptions()
474 op->weights_format = FullyConnectedWeightsFormat::kDefault; in ReadOptions()
477 op->weights_format = FullyConnectedWeightsFormat::kShuffled4x16Int8; in ReadOptions()
481 op->weights_format = FullyConnectedWeightsFormat::kDefault; in ReadOptions()
491 op_sig.options.fully_connected.weights_format = in GetVersion()
492 GetWeightFormat(fc_op.weights_format); in GetVersion()
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected_test.cc142 FullyConnectedOptionsWeightsFormat weights_format = argument
179 if (weights_format != FullyConnectedOptionsWeightsFormat_DEFAULT) {
186 weights_format, keep_num_dims)
701 int batches, FullyConnectedOptionsWeightsFormat weights_format) { in SimpleTestQuantizedInt16OutputCase() argument
720 /*activation_func=*/ActivationFunctionType_NONE, weights_format); in SimpleTestQuantizedInt16OutputCase()
735 switch (weights_format) { in SimpleTestQuantizedInt16OutputCase()
Dfully_connected.cc137 is_quantized && (params->weights_format == in CheckTypes()
197 params->weights_format == kTfLiteFullyConnectedWeightsFormatDefault ? 1 in PrepareImpl()
932 if (params->weights_format == in Eval()
941 } else if (params->weights_format == in Eval()
951 if (params->weights_format == kTfLiteFullyConnectedWeightsFormatDefault) { in Eval()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Doptimize.mlir284 …, %cst1) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
291 …NSTANT0]]) {fused_activation_function = "RELU6", keep_num_dims = false, weights_format = "DEFAULT"}
301 fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"
320 …, %cst1) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
327 … %[[b]]) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
337 …, %cst1) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
342 … %[[b]]) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
351 …1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
364 …, %cst1) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
369 … %[[b]]) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
[all …]
Dconst-fold.mlir736 …st_bias) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
750 …st_bias) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
765 …st_bias) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
780 …st_bias) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
786 …CST_1]]) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
796 …st_bias) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
802 …CST_1]]) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
812 …st_bias) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
Dquantize.mlir107 …4, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
113 …%[[cst_0]]) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"}
121 …, %arg2) {fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"} :…
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dfully_connected_op_builder.cc152 if (fc_params->weights_format != kTfLiteFullyConnectedWeightsFormatDefault) { in IsFullyConnectedOpSupported()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h177 TfLiteFullyConnectedWeightsFormat weights_format; member
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc395 /*weights_format=*/rewriter.getStringAttr(fc_op.weights_format()), in matchAndRewrite()
418 rewriter.getStringAttr(fully_connected_op.weights_format()); in matchAndRewrite()
503 /*weights_format=*/rewriter.getStringAttr(fc_op.weights_format()), in matchAndRewrite()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc221 matmul_params->weights_format == in IsNodeSupportedByHexagon()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc1280 switch (schema_params->weights_format()) { in ParseFullyConnected()
1282 params->weights_format = kTfLiteFullyConnectedWeightsFormatDefault; in ParseFullyConnected()
1285 params->weights_format = in ParseFullyConnected()

12