Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for CreateEqualSizeSplitVOp (0.22 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/lite/utils/lstm_utils.cc 

      Operation* weights_array;
  if (failed(CreateEqualSizeSplitVOp(transposed_weight_kernel, 0, splits,
                                     func_op.getLoc(), builder,
                                     &weights_array)))
    return failure();

  // Splits the recurrent_weights into 4:
  Operation* recurrent_weights_array;
  if (failed(CreateEqualSizeSplitVOp(transpose_recurrent_kernel, 0, splits,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 36.2K bytes
    - Viewed (0)
Back to top