Home
last modified time | relevance | path

Searched refs:unidirectional_sequence_lstm (Results 1 – 25 of 49) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
H A Dsplit-merged-operands.mlir7 …// CHECK: %[[LSTM:[a-z0-9]*]] = "tfl.unidirectional_sequence_lstm"(%arg2, %arg0, %arg0, %arg0, %a…
10 …%1 = "tfl.unidirectional_sequence_lstm"(%arg2, %arg0, %arg0, %arg0, %arg0, %arg0, %arg0, %arg0, %a…
18 …// CHECK: %[[LSTM_1:[a-z0-9]*]] = "tfl.unidirectional_sequence_lstm"(%arg2, %arg0, %arg0, %arg0, …
21 …// CHECK: %[[LSTM_2:[a-z0-9]*]] = "tfl.unidirectional_sequence_lstm"(%[[LSTM_1]], %arg0, %arg0, %…
24 …%1 = "tfl.unidirectional_sequence_lstm"(%arg2, %arg0, %arg0, %arg0, %arg0, %arg0, %arg0, %arg0, %a…
25 …%2 = "tfl.unidirectional_sequence_lstm"(%1, %arg0, %arg0, %arg0, %arg0, %arg0, %arg0, %arg0, %arg0…
H A Dprepare-quantize-post-training.mlir14 %0 = "tfl.unidirectional_sequence_lstm"(%arg0,
37 // CHECK: %[[lstm:.*]] = "tfl.unidirectional_sequence_lstm"(%arg0, {{(%[^%,]+, )+}}%[[dq]], %[[none…
69 %23 = "tfl.unidirectional_sequence_lstm"(%input,
112 // CHECK: %[[lstm:.*]] = "tfl.unidirectional_sequence_lstm"(%[[input_0]], %[[input_1]], %[[input_2]…
229 %23 = "tfl.unidirectional_sequence_lstm"(%input,
280 // CHECK: %[[lstm:.*]] = "tfl.unidirectional_sequence_lstm"(%[[input_0]], %[[input_1]], %[[input_2]…
320 %23 = "tfl.unidirectional_sequence_lstm"(%input,
374 %23 = "tfl.unidirectional_sequence_lstm"(%arg0,
H A Dshape-inference.mlir82 …// CHECK: "tfl.unidirectional_sequence_lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg…
83 …%0 = "tfl.unidirectional_sequence_lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg7, %a…
H A Dprepare-composite-functions-tf.mlir207 // CHECK: [[VAL_20:%.*]] = "tfl.unidirectional_sequence_lstm"([[VAL_0]], [[VAL_12]]#0, [[…
248 // CHECK: [[VAL_20:%.*]] = "tfl.unidirectional_sequence_lstm"([[VAL_0]], [[VAL_12]]#0, [[…
292 // CHECK: [[VAL_22:%.*]] = "tfl.unidirectional_sequence_lstm"([[VAL_7]], [[VAL_14]]#0, [[…
336 // CHECK: [[VAL_22:%.*]] = "tfl.unidirectional_sequence_lstm"([[VAL_7]], [[VAL_14]]#0, [[…
385 // CHECK: [[VAL_20:%.*]] = "tfl.unidirectional_sequence_lstm"([[VAL_0]], [[VAL_12]]#0, [[…
435 // CHECK: [[VAL_20:%.*]] = "tfl.unidirectional_sequence_lstm"([[VAL_0]], [[VAL_12]]#0, [[…
H A Dops.mlir733 …// CHECK: "tfl.unidirectional_sequence_lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg…
734 …%0 = "tfl.unidirectional_sequence_lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg7, %a…
742 …// CHECK: "tfl.unidirectional_sequence_lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg…
743 …%0 = "tfl.unidirectional_sequence_lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg7, %a…
758 // CHECK: "tfl.unidirectional_sequence_lstm"(%arg0,
765 %0 = "tfl.unidirectional_sequence_lstm"(%arg0,
779 …// CHECK: "tfl.unidirectional_sequence_lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg…
780 …%0 = "tfl.unidirectional_sequence_lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg7, %a…
788 …// expected-error @+1 {{'tfl.unidirectional_sequence_lstm' op failed to verify that either project…
789 …%0 = "tfl.unidirectional_sequence_lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg7, %a…
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/src/
DUnidirectionalSequenceLSTM.cpp24 namespace unidirectional_sequence_lstm { namespace
82 NN_DEFINE_VALIDATION_FUNCTION(UNIDIRECTIONAL_SEQUENCE_LSTM, unidirectional_sequence_lstm::validate);
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceLSTM.cpp35 namespace unidirectional_sequence_lstm { namespace
417 unidirectional_sequence_lstm::prepare,
418 unidirectional_sequence_lstm::execute,
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunidirectional_sequence_lstm.cc36 namespace unidirectional_sequence_lstm { namespace
1454 static TfLiteRegistration r = {unidirectional_sequence_lstm::Init, in Register_UNIDIRECTIONAL_SEQUENCE_LSTM()
1455 unidirectional_sequence_lstm::Free, in Register_UNIDIRECTIONAL_SEQUENCE_LSTM()
1456 unidirectional_sequence_lstm::Prepare, in Register_UNIDIRECTIONAL_SEQUENCE_LSTM()
1457 unidirectional_sequence_lstm::Eval}; in Register_UNIDIRECTIONAL_SEQUENCE_LSTM()
H A DAndroid.bp168 "unidirectional_sequence_lstm.cc",
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/
DUnidirectionalSequenceLSTM.h22 namespace android::nn::unidirectional_sequence_lstm {
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
H A Dunidirectional_sequence_lstm.mlir232 // CHECK-NEXT: name: "tfl.unidirectional_sequence_lstm",
324 %2 = "tfl.unidirectional_sequence_lstm"(%arg0,
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
H A DBUILD92 "//tensorflow/lite:testdata/unidirectional_sequence_lstm.bin",
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/
H A Dbuild_def.bzl17 "unidirectional_sequence_lstm",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
H A Dlstm.mlir35 …// CHECK: "tfl.unidirectional_sequence_lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg…
36 …%0 = "tfl.unidirectional_sequence_lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg7, %a…
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dtflite_static.bp163 "kernels/unidirectional_sequence_lstm.cc",
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/
H A Dbuild_def.bzl199 "unidirectional_sequence_lstm",
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/libtflite_static/linux_glibc_x86_64_static/
Dmeta_lic.rsp1 …nspose_conv.cc -s external/tensorflow/tensorflow/lite/kernels/unidirectional_sequence_lstm.cc -s e…
Dmeta_lic158 sources: "external/tensorflow/tensorflow/lite/kernels/unidirectional_sequence_lstm.cc"
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/libtflite_static/android_arm64_armv8-2a_cortex-a55_static_apex33/
Dmeta_lic158 sources: "external/tensorflow/tensorflow/lite/kernels/unidirectional_sequence_lstm.cc"
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/libtflite_static/android_arm64_armv8-2a_cortex-a55_static/
Dmeta_lic158 sources: "external/tensorflow/tensorflow/lite/kernels/unidirectional_sequence_lstm.cc"
Dmeta_lic.rsp1 …nspose_conv.cc -s external/tensorflow/tensorflow/lite/kernels/unidirectional_sequence_lstm.cc -s e…
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/libtflite_static/android_arm64_armv8-2a_cortex-a55_static_apex30/
Dmeta_lic158 sources: "external/tensorflow/tensorflow/lite/kernels/unidirectional_sequence_lstm.cc"
Dmeta_lic.rsp1 …nspose_conv.cc -s external/tensorflow/tensorflow/lite/kernels/unidirectional_sequence_lstm.cc -s e…
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/libtflite_static/android_arm64_armv8-2a_cortex-a55_sdk_static_apex30/
Dmeta_lic158 sources: "external/tensorflow/tensorflow/lite/kernels/unidirectional_sequence_lstm.cc"
Dmeta_lic.rsp1 …nspose_conv.cc -s external/tensorflow/tensorflow/lite/kernels/unidirectional_sequence_lstm.cc -s e…

12