Home
last modified time | relevance | path

Searched refs:kTfLiteFunctionInputIndex (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_ophint_func_op.cc42 constexpr char kTfLiteFunctionInputIndex[] = "_tflite_function_input_index"; variable
127 auto input_index_attr = composite_func_op.getAttr(kTfLiteFunctionInputIndex) in BuildUnidirectionalSequenceLSTMOp()
Dextract_ophint.cc58 constexpr char kTfLiteFunctionInputIndex[] = "_tflite_function_input_index"; variable
536 kTfLiteFunctionInputIndex, builder->getI32ArrayAttr(input_indexes))); in BuildFusedFuncOp()
573 op.getAttrOfType<IntegerAttr>(kTfLiteFunctionInputIndex); in FindAllOphintNodes()