Searched refs:CreateEqualSizeSplitVOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 591 LogicalResult CreateEqualSizeSplitVOp(Value input, int axis, int splits, in CreateEqualSizeSplitVOp() function
691 if (failed(CreateEqualSizeSplitVOp(transposed_weight_kernel, 0, splits, in ConvertKerasLSTMLayer()
698 if (failed(CreateEqualSizeSplitVOp(transpose_recurrent_kernel, 0, splits, in ConvertKerasLSTMLayer()
705 if (failed(CreateEqualSizeSplitVOp(bias, 0, splits, func_op.getLoc(), builder, in ConvertKerasLSTMLayer()
|