Home
last modified time | relevance | path

Searched refs:keep_num_dims (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dimporter_test_min_max.cc.mlir7 …y_connected"(%arg0, %arg1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
12 …ected"(%arg0, %arg1, %[[CONST]]) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
Dexternal_constant.mlir7 …y_connected"(%arg0, %arg1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
12 …ected"(%arg0, %arg1, %[[CONST]]) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
Doptional.mlir9 …y_connected"(%arg0, %arg1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/
Dunroll_batch_matmul.pbtxt97 …L_11]], %[[VAL_17]], %[[VAL_5]]) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
98 …L_13]], %[[VAL_17]], %[[VAL_5]]) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.h39 bool keep_num_dims; member
Dop_version.cc147 if (op_sig.options.fully_connected.keep_num_dims) { in GetBuiltinOperatorVersion()
712 op_sig.options.fully_connected.keep_num_dims = in GetOpSignature()
713 fully_connected_option->keep_num_dims(); in GetOpSignature()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Doptional.mlir5 …%1:2 = "tfl.fully_connected"(%arg0, %arg1, %0) {fused_activation_function = "NONE", keep_num_dims …
Dsignature_def.mlir113 …_connected"(%arg0, %cst_0, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
114 …_connected"(%arg0, %cst_1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
Dfully_connected_v2.mlir76 …y_connected"(%arg0, %arg1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
Dfully_connected.mlir76 …y_connected"(%arg0, %arg1, %cst) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Doptimize.mlir284 …_connected"(%arg0, %cst0, %cst1) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
291 … %[[CONSTANT]], %[[CONSTANT0]]) {fused_activation_function = "RELU6", keep_num_dims = false, weigh…
301 fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"
320 …%1 = "tfl.fully_connected"(%0, %dq, %cst1) {fused_activation_function = "NONE", keep_num_dims = fa…
327 …nnected"(%arg0, %[[dq]], %[[b]]) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
337 …%1 = "tfl.fully_connected"(%0, %cst0, %cst1) {fused_activation_function = "NONE", keep_num_dims = …
342 …onnected"(%arg0, %[[w]], %[[b]]) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
351 …%1 = "tfl.fully_connected"(%0, %arg1, %cst) {fused_activation_function = "NONE", keep_num_dims = f…
364 …%1 = "tfl.fully_connected"(%0, %cst0, %cst1) {fused_activation_function = "NONE", keep_num_dims = …
369 …onnected"(%arg0, %[[w]], %[[b]]) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
[all …]
Dconst-fold.mlir736 …_input, %cst_weights, %cst_bias) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
750 …_input, %cst_weights, %cst_bias) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
765 …_input, %cst_weights, %cst_bias) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
780 …_input, %cst_weights, %cst_bias) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
786 …[[CST]], %[[CST_0]], %[[CST_1]]) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
796 …_input, %cst_weights, %cst_bias) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
802 …[[CST]], %[[CST_0]], %[[CST_1]]) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
812 …_input, %cst_weights, %cst_bias) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
Dquantize.mlir107 …%5 = "tfl.fully_connected"(%2, %4, %cst) {fused_activation_function = "NONE", keep_num_dims = fals…
113 …"(%arg0, %[[cst_1]], %[[cst_0]]) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
121 …%2 = "tfl.fully_connected"(%0, %1, %arg2) {fused_activation_function = "NONE", keep_num_dims = fal…
Dpost-quantize.mlir24 …_connected"(%arg0, %arg1, %arg2) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
Dprepare-quantize-signed.mlir172 …%fc = "tfl.fully_connected"(%arg0, %w, %b) {fused_activation_function = "NONE", keep_num_dims = fa…
Dtfl_while_outline.mlir114 …_connected"(%8, %cst_10, %cst_0) {fused_activation_function = "NONE", keep_num_dims = false, weigh…
Dcanonicalize.mlir211 …%1 = "tfl.fully_connected"(%arg0, %arg1, %0) {fused_activation_function = "NONE", keep_num_dims = …
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h183 bool keep_num_dims; member
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc396 /*keep_num_dims=*/rewriter.getBoolAttr(fc_op.keep_num_dims())); in matchAndRewrite()
420 rewriter.getBoolAttr(fully_connected_op.keep_num_dims()); in matchAndRewrite()
504 /*keep_num_dims=*/rewriter.getBoolAttr(fc_op.keep_num_dims())); in matchAndRewrite()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc220 matmul_params->keep_num_dims == false && in IsNodeSupportedByHexagon()
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected_test.cc140 bool keep_num_dims = false, bool bias_tensor_optional = false, argument
186 weights_format, keep_num_dims)
Dfully_connected.cc352 if (params->keep_num_dims) { in PrepareImpl()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc1276 params->keep_num_dims = schema_params->keep_num_dims(); in ParseFullyConnected()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h569 bool keep_num_dims = false;
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc490 op_sig.options.fully_connected.keep_num_dims = fc_op.keep_num_dims; in GetVersion()

12