Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc634 Value weight_kernel = func_op.getArgument(3); in ConvertKerasLSTMLayer() local
673 weight_kernel.getType().cast<RankedTensorType>(); in ConvertKerasLSTMLayer()
678 Transpose2D(builder, weight_kernel, weight_type, func_op.getLoc()); in ConvertKerasLSTMLayer()